bzoj3998 [TJOI2015]弦论

地址：http://www.lydsy.com/JudgeOnline/problem.php?id=3998

题面：

3998: [TJOI2015]弦论

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 3127  Solved: 1079

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^5

T<2

K<=10^9

 

思路：　　

　　对于t=1的情况，把每个状态的cnt记为1.

　　t=0时，把cnt记为endpos集合的大小。

　　然后dp即可。

/**************************************************************
    Problem: 3998
    User: weeping
    Language: C++
    Result: Accepted
    Time:6736 ms
    Memory:122876 kb
****************************************************************/
 
#include <bits/stdc++.h>
 
using namespace std;
 
 
 
struct SAM
{
    static const int MAXN = 500002<<1;//大小为字符串长度两倍
    static const int LetterSize = 26;
 
    int tot, last, ch[MAXN][LetterSize], fa[MAXN], len[MAXN];
    int sum[MAXN], tp[MAXN], cnt[MAXN]; //sum,tp用于拓扑排序，tp为排序后的数组
 
    void init( void)
    {
        last = tot = 1;
        len[1] = 0;
        memset(ch,0,sizeof ch);
        memset(fa,0,sizeof fa);
        memset(cnt,0,sizeof cnt);
    }
 
    void add( int x)
    {
        int p = last, np = last = ++tot;
        len[np] = len[p] + 1, cnt[last] = 1;
        while( p && !ch[p][x]) ch[p][x] = np, p = fa[p];
        if( p == 0)
            fa[np] = 1;
        else
        {
            int q = ch[p][x];
            if( len[q] == len[p] + 1)
                fa[np] = q;
            else
            {
                int nq = ++tot;
                memcpy( ch[nq], ch[q], sizeof ch[q]);
                len[nq] = len[p] + 1, fa[nq] = fa[q], fa[q] = fa[np] = nq;
                while( p && ch[p][x] == q)  ch[p][x] = nq, p = fa[p];
            }
        }
    }
 
    void toposort( void)
    {
        for(int i = 1; i <= len[last]; i++)   sum[i] = 0;
        for(int i = 1; i <= tot; i++)   sum[len[i]]++;
        for(int i = 1; i <= len[last]; i++)   sum[i] += sum[i-1];
        for(int i = 1; i <= tot; i++)   tp[sum[len[i]]--] = i;
        for(int i = tot; i; i--)   cnt[fa[tp[i]]] += cnt[tp[i]];
    }
 
    void solve( int t, int k)
    {
        for(int i = 1; i <= len[last]; i++)   sum[i] = 0;
        for(int i = 1; i <= tot; i++)   sum[len[i]]++;
        for(int i = 1; i <= len[last]; i++)   sum[i] += sum[i-1];
        for(int i = 1; i <= tot; i++)   tp[sum[len[i]]--] = i;
        for(int i = tot; i; i--)
        if(t)   cnt[fa[tp[i]]] += cnt[tp[i]];
        else cnt[i] = 1;
        cnt[1]=0;
        for(int i = 1; i <= tot; i++)   sum[i] = cnt[i];
        for(int i = tot; i; i--)
        for(int j = 0; j < 26; j++)
            sum[tp[i]] += sum[ch[tp[i]][j]];
        if(sum[1]<k) {printf("-1
");return;}
        int p=1;
        while(k>0)
        {
            for(int j = 0; j < 26; j++)
            if(sum[ch[p][j]]>=k)
            {
                printf("%c",'a'+j),p=ch[p][j],k-=cnt[p];
                break;
            }
            else
                k-=sum[ch[p][j]];
        }
    }
} sam;
 
char ss[500004];
 
int main(void)
{
    int t,k;
    //freopen("in.acm","r",stdin);
    sam.init();
    scanf("%s%d%d",ss,&t,&k);
    for(int i=0,len=strlen(ss);i<len;i++)  sam.add(ss[i]-'a');
    sam.solve(t,k);
    return 0;
}