[BZOJ]3998 [TJOI2015] 弦论 后缀自动机

3998: [TJOI2015]弦论

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 3292  Solved: 1133
[Submit][Status][Discuss]

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

Source

[Submit][Status][Discuss]

HOME Back

　　以前wjj给我讲过但那是暑假的事了... 现在几乎都忘光了... 不过重拾起来还是非常快? 虽然一开始看毛子论文的各种专业术语乱飚一脸懵逼， 不过努力回想wjj讲过的内容在纸上推推还是推出来了... 明白了以后看了下clj神犇的论文发现这玩意性质还非常多? 感觉比今早上学的回文自动机要高级多了... 看来还是得多学习一个... 争取深入理解后缀自动机， 把性质都记录下来推一推多做点题.

　　这道题很简单, 两种情况分类讨论逆拓扑序倒推预处理, 然后询问第k大可以说是...26分?

#include<stdio.h>#include<cstring>#include<algorithm>using namespace std;const int maxn = 1000005;char ss[maxn];int n, k, T, cnt = 1, last = 1;int ws[maxn], sa[maxn], c[maxn][26], dep[maxn], val[maxn], sum[maxn], par[maxn];struct sam{inline void build(int x) {int p = last, np = last = ++ cnt;dep[np] = dep[p] + 1; val[np] = 1;while (p && !c[p][x]) c[p][x] = np, p = par[p];if (!p) par[np] = 1;else {int who = c[p][x];if (dep[who] == dep[p] + 1) par[np] = who;else {int anwho = ++ cnt;dep[anwho] = dep[p] + 1;memcpy(c[anwho], c[who], sizeof(c[who]));par[anwho] = par[who];  par[np] = par[who] = anwho;  while (c[p][x] == who) c[p][x] = anwho, p = par[p];  }}}inline void init() {for (int i = 1; i <= cnt; ++ i) ws[dep[i]] ++;  for (int i = 1; i <= n; ++ i) ws[i] += ws[i - 1];for (int i = cnt; i; -- i) sa[ws[dep[i]]--] = i;for (int i = cnt; i; -- i) {int t = sa[i];if (T & 1) val[par[t]] += val[t];else val[t] = 1;}val[1] = 0;for (int i = cnt; i; -- i) {int t = sa[i]; sum[t] = val[t];for (int ch = 0; ch < 26; ++ ch)  sum[t] += sum[c[t][ch]];}}void calc(int x,int k) {  while(true) {if (k <= val[x]) return;k -= val[x];for (int i = 0; i < 26; ++ i) {  int t = c[x][i];  if (t) {  if (k <= sum[t]) {  putchar(i + 'a');  x = t;  break;}k -= sum[t];}}}    }}sam;  int main() {scanf("%s", ss + 1); n = strlen(ss + 1);scanf("%d%d", &T, &k);for (int i = 1; i <= n; ++ i) sam.build(ss[i] - 'a');sam.init();if (k > sum[1]) puts("-1");else sam.calc(1,k);return 0;}