【BZOJ3998】【TJOI2015】弦论 后缀自动机

链接:

#include <stdio.h>
int main()
{
    puts("转载请注明出处[vmurder]谢谢");
    puts("网址:blog.csdn.net/vmurder/article/details/45369569");
}

题解:

首先我们可以建一个后缀自动机。

然后每条路径走到每个点都是一个串,它们是有字典序的。

我们只需要统计出往每个点走之后都有多少串就好了。

fi=(∑fson)+numi

对于不计重复的情况下,numi=1

对于计算重复的情况下,每个节点都有多种走到最后的方式,numi 就是看有这个种数。

比如 ababab,aba 这个串最后可以走成 abab、ababab 。于是 num 是2。

初值是所有 np 的 num=1 ,然后 numi=∑numson 。

f函数 处理出来以后,就随便搞了。

每次往下走就行了,类似26分? (雾

不妨看我的 dfs 函数。

代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 1001000
#define T 26
#define inf 0x3f3f3f3f
using namespace std;
struct SAM
{
    int son[N][T],pa[N],dep[N],last,cnt;
    int val[N],f[N];
    void init(){last=cnt=1;}
    int newnode(int p){dep[++cnt]=dep[p]+1;return cnt;}
    void add(int x)
    {
        int p=last;
        int np=newnode(p);
        while(p&&!son[p][x])son[p][x]=np,p=pa[p];
        if(!p)pa[np]=1;
        else {
            int q=son[p][x];
            if(dep[q]==dep[p]+1)pa[np]=q;
            else {
                int nq=newnode(p);
                pa[nq]=pa[q],pa[q]=pa[np]=nq;
                memcpy(son[nq],son[q],sizeof son[q]);
                while(p&&son[p][x]==q)son[p][x]=nq,p=pa[p];
            }
        }
        val[last=np]=1;
    }
    int d[N],stk[N],top;
    struct Eli
    {
        int v,next;
    }e[N*T/5];
    int head[N],ct;
    void add(int u,int v)
    {
        d[v]++;
        e[++ct].v=v;
        e[ct].next=head[u];
        head[u]=ct;
    }
    void bfsf()
    {
        int i,j,u,v;
        for(i=1;i<=cnt;i++)for(j=0;j<T;j++)if(son[i][j])
                add(son[i][j],i);
        for(i=1;i<=cnt;i++)if(!d[i])stk[++top]=i;
        while(top)
        {
            u=stk[top--],f[u]++,val[u]=1;
            for(i=head[u];i;i=e[i].next)
            {
                v=e[i].v,f[v]+=f[u];
                if(!--d[v])stk[++top]=v;
            }
        }
        f[1]--;
    }
    void bfsg()
    {
        int i,j,u,v;
        for(i=2;i<=cnt;i++)d[pa[i]]++;
        for(i=1;i<=cnt;i++)if(!d[i])stk[++top]=i;
        while(top)
        {
            u=stk[top--];
            val[pa[u]]+=val[u];
            if(!--d[pa[u]])stk[++top]=pa[u];
        }
        for(i=1;i<=cnt;i++)for(j=0;j<T;j++)
            if(son[i][j])add(son[i][j],i);
        for(i=1;i<=cnt;i++)if(!d[i])stk[++top]=i;
        while(top)
        {
            u=stk[top--],f[u]+=val[u];
            for(i=head[u];i;i=e[i].next)
            {
                v=e[i].v,f[v]+=f[u];
                if(!--d[v])stk[++top]=v;
            }
        }
        f[1]-=val[1];
    }
    void dfs(int x,int k)
    {
        int i,v;
        for(i=0;i<T;i++)if(v=son[x][i])
        {
            if(k<=f[v])
            {
                printf("%c",‘a‘+i);
                k-=val[v];
                if(k>0)dfs(v,k);
                return ;
            }
            else k-=f[v];
        }
    }
}sam;
char s[N/2];
int main()
{
    freopen("test.in","r",stdin);
    int i,j,k;

    scanf("%s",s+1);
    sam.init();
    for(i=1;s[i];i++)sam.add(s[i]-‘a‘);
    scanf("%d%d",&j,&k);
    if(j==0)sam.bfsf();
    else sam.bfsg();
    sam.dfs(1,k);
    return 0;
}
时间: 2024-08-26 20:32:58

【BZOJ3998】【TJOI2015】弦论 后缀自动机的相关文章

[bzoj3998][TJOI2015]弦论——后缀自动机

题目大意: 给定一个字符串,求它的第k小子串. 思路: 后缀自动机的模板题. 考虑将后缀自动机建出来之后,求出每一个状态在原串中出现了多少次,然后统计以每个节点往后拓展的字符串总共有多少种(按照拓扑序累加即可),最后直接在DAG上做类似二分的就好了. /*======================================= * Author : ylsoi * Time : 2019.2.9 * Problem : bzoj3998 * E-mail : [email protecte

BZOJ3998 TJOI2015 弦论 后缀自动机

题意:求一个字符串的第K小字串,T=0表示不同位置相同的子串算作一个,T=1算作多个 题意: 建出SAM来跑第K子串,由于一个点所代表的子串在原串出现次数为其子树叶子结点的数量,因而有: T==1,每个点的|right|=1 T==2,每个点的|right|=子树叶子结点数 BFS跑出所有子串出现的次数即|right|,DFS统计每个节点的出现次数和,具体细节看代码. 然后处理询问即可. #include <cstdio> #include <cstring> #include &

【BZOJ3998】[TJOI2015]弦论 后缀自动机

[BZOJ3998][TJOI2015]弦论 Description 对于一个给定长度为N的字符串,求它的第K小子串是什么. Input 第一行是一个仅由小写英文字母构成的字符串S 第二行为两个整数T和K,T为0则表示不同位置的相同子串算作一个.T=1则表示不同位置的相同子串算作多个.K的意义如题所述. Output 输出仅一行,为一个数字串,为第K小的子串.如果子串数目不足K个,则输出-1 Sample Input aabc 0 3 Sample Output aab HINT N<=5*10

【bzoj3998】[TJOI2015]弦论 后缀自动机+dp

题目描述 对于一个给定长度为N的字符串,求它的第K小子串是什么. 输入 第一行是一个仅由小写英文字母构成的字符串S 第二行为两个整数T和K,T为0则表示不同位置的相同子串算作一个.T=1则表示不同位置的相同子串算作多个.K的意义如题所述. 输出 输出仅一行,为一个数字串,为第K小的子串.如果子串数目不足K个,则输出-1 样例输入 aabc 0 3 样例输出 aab 题解 后缀自动机+dp 先对原串建立后缀自动机,然后在其上面跑dp统计每个节点开始的串的个数. 设f[i]表示与位置i有相同前缀的串

bzoj 3998: [TJOI2015]弦论(后缀自动机)

题目链接:bzoj 3998: [TJOI2015]弦论 题意: 对于一个给定长度为N的字符串,求它的第K小子串是什么. 题解: 后缀自动机O(n)*26解决. 对于op=0,num[i]=1,对于op=1,num[i]=cnt[i]. 因为cnt[i](即right集)表示以i节点结尾的后缀出现的次数. 1 #include<cstdio> 2 #include<cstring> 3 #define F(i,a,b) for(int i=a;i<=b;++i) 4 #def

【BZOJ-3998】弦论 后缀自动机

3998: [TJOI2015]弦论 Time Limit: 10 Sec  Memory Limit: 256 MBSubmit: 2018  Solved: 662[Submit][Status][Discuss] Description 对于一个给定长度为N的字符串,求它的第K小子串是什么. Input 第一行是一个仅由小写英文字母构成的字符串S 第二行为两个整数T和K,T为0则表示不同位置的相同子串算作一个.T=1则表示不同位置的相同子串算作多个.K的意义如题所述. Output 输出仅

[TJOI2015]弦论 - 后缀自动机

下了狠心开始做SAM的题目了-- (中间因为傻逼26分写错被卡,进来的时候记得把自己的 cnt 减掉) // TJOI2015 XIAN LUN #include <bits/stdc++.h> using namespace std; const int Maxn = 2000005; struct Suffix_Automata { int maxlen[Maxn], trans[Maxn][26], link[Maxn], Size, Last; int t[Maxn], a[Maxn]

BZOJ3998 [TJOI2015]弦论 【后缀自动机】

题目 对于一个给定长度为N的字符串,求它的第K小子串是什么. 输入格式 第一行是一个仅由小写英文字母构成的字符串S 第二行为两个整数T和K,T为0则表示不同位置的相同子串算作一个.T=1则表示不同位置的相同子串算作多个.K的意义如题所述. 输出格式 输出仅一行,为一个数字串,为第K小的子串.如果子串数目不足K个,则输出-1 输入样例 aabc 0 3 输出样例 aab 提示 N<=5*10^5 T<2 K<=10^9 题解 肝了一个中午的论文还是想了好久这种裸题.. 由后缀自动机从根节点

Bzoj3998 [TJOI2015]弦论

Time Limit: 10 Sec  Memory Limit: 256 MBSubmit: 2388  Solved: 798 Description 对于一个给定长度为N的字符串,求它的第K小子串是什么. Input 第一行是一个仅由小写英文字母构成的字符串S 第二行为两个整数T和K,T为0则表示不同位置的相同子串算作一个.T=1则表示不同位置的相同子串算作多个.K的意义如题所述. Output 输出仅一行,为一个数字串,为第K小的子串.如果子串数目不足K个,则输出-1 Sample In