题目描述
对于一个给定长度为N的字符串，求它的第K小子串是什么。
输入
第一行是一个仅由小写英文字母构成的字符串S
第二行为两个整数T和K，T为0则表示不同位置的相同子串算作一个。T=1则表示不同位置的相同子串算作多个。K的意义如题所述。
输出
输出仅一行，为一个数字串，为第K小的子串。如果子串数目不足K个，则输出-1
样例输入
aabc
0 3
样例输出
aab
题解
后缀自动机+dp
先对原串建立后缀自动机，然后在其上面跑dp统计每个节点开始的串的个数。
设f[i]表示与位置i有相同前缀的串的个数。
那么当T=0时，显然f[i]=∑f[son[i]]+1。
当T=1时，f[i]=∑f[son[i]]+|right[i]|，需要统计right集合的大小，也即统计parent树中子树内有多少个叶子结点，这个递推一下即可。
在这里边需要保证son[i]在i之前更新，所以需要得到拓扑序。
然后大爷说会卡常？这里orz hzwer，对dis排序即可得到拓扑序，而且可以使用基数排序，详见代码。
最后求一下和，dfs一遍就好了，类似于二（十六）分。
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 1000010
using namespace std;
int n , opt , next[N][26] , fa[N] , dis[N] , last = 1 , tot = 1 , v[N] , q[N] , cnt[N] , sum[N];
char str[N];
void ins(int c)
{
int p = last , np = last = ++tot;
dis[np] = dis[p] + 1 , cnt[np] = 1;
while(p && !next[p][c]) next[p][c] = np , p = fa[p];
if(!p) fa[np] = 1;
else
{
int q = next[p][c];
if(dis[q] == dis[p] + 1) fa[np] = q;
else
{
int nq = ++tot;
memcpy(next[nq] , next[q] , sizeof(next[q])) , dis[nq] = dis[p] + 1 , fa[nq] = fa[q] , fa[np] = fa[q] = nq;
while(p && next[p][c] == q) next[p][c] = nq , p = fa[p];
}
}
}
void init()
{
int i , j , t;
for(i = 1 ; i <= tot ; i ++ ) v[dis[i]] ++ ;
for(i = 1 ; i <= n ; i ++ ) v[i] += v[i - 1];
for(i = tot ; i ; i -- ) q[v[dis[i]] -- ] = i;
for(i = tot ; i ; i -- )
{
t = q[i];
if(opt) cnt[fa[t]] += cnt[t];
else cnt[t] = 1;
}
cnt[1] = 0;
for(i = tot ; i ; i -- )
{
t = q[i] , sum[t] = cnt[t];
for(j = 0 ; j < 26 ; j ++ )
sum[t] += sum[next[t][j]];
}
}
void query(int p , int k)
{
if(k <= cnt[p]) return;
k -= cnt[p];
int i;
for(i = 0 ; i < 26 ; i ++ )
{
if(next[p][i])
{
if(k <= sum[next[p][i]])
{
putchar(i + 'a') , query(next[p][i] , k);
return;
}
k -= sum[next[p][i]];
}
}
}
int main()
{
int k , i;
scanf("%s%d%d" , str + 1 , &opt , &k) , n = strlen(str + 1);
for(i = 1 ; i <= n ; i ++ ) ins(str[i] - 'a');
init();
if(k > sum[1]) printf("-1");
else query(1 , k);
return 0;
}