BZOJ3473 字符串 【廣義後綴自動機】
阿新 • • 發佈:2018-03-10
tchar const for return AR pri 題意 沒有 long
題目
給定n個字符串,詢問每個字符串有多少子串(不包括空串)是所有n個字符串中至少k個字符串的子串?
輸入格式
第一行兩個整數n,k。
接下來n行每行一個字符串。
輸出格式
一行n個整數,第i個整數表示第i個字符串的答案。
輸入樣例
3 1
abc
a
ab
輸出樣例
6 1 3
提示
對於 100% 的數據,1<=n,k<=10^5,所有字符串總長不超過10^5,字符串只包含小寫字母。
題解
我們先建一個廣義後綴自動機
然後用每個原串在SAM上走,走到的節點就是parent樹的葉子節點,將其沿parent邊一直往上+1,表示這些節點表示的字符串+1【當然只能加一次,所以再開一個數組cur[]表示加過沒有】
處理完後,我們就可以統計答案了
先拓撲排序,設f[]為該位置滿足題意的字符串個數,顯然如果一個位置累加的字符串>=k,那麽該位置表示的至少為\(step[u] - step[pre[u]]\)
但還不完全,父親的代表的字符串個數同樣也符合該點
然後每個串再在SAM上走一遍統計答案即可
#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#include<algorithm>
#define LL long long int
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<‘ ‘; puts("");
using namespace std;
const int maxn = 200005,maxm = 100005,INF = 1000000000;
inline int read(){
int out = 0,flag = 1; char c = getchar();
while (c < 48 || c > 57){if (c == ‘-‘) flag = -1; c = getchar();}
while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
return out * flag;
}
string s[maxn];
char ss[maxn];
int ch[maxn][26],pre[maxn],step[maxn],sz[maxn],cnt,last,n,k,cur[maxn];
LL f[maxn];
int ins(int x){
int p = last,np = ++cnt; step[np] = step[p] + 1; last = np;
while (p && !ch[p][x]) ch[p][x] = np,p = pre[p];
if (!p) pre[np] = 1;
else {
int q = ch[p][x];
if (step[q] == step[p] + 1) pre[np] = q;
else {
int nq = ++cnt; step[nq] = step[p] + 1;
for (int i = 0; i < 26; i++) ch[nq][i] = ch[q][i];
pre[nq] = pre[q]; pre[np] = pre[q] = nq;
while (ch[p][x] == q) ch[p][x] = nq,p = pre[p];
}
}
return np;
}
int b[maxn],a[maxn];
LL ans;
void tsort(){
REP(i,cnt) b[step[i]]++;
REP(i,cnt) b[i] += b[i - 1];
for (int i = cnt; i; i--) a[b[step[i]]--] = i;
}
int main(){
n = read(); k = read();
last = cnt = 1;
for (int i = 1; i <= n; i++){
last = 1;
scanf("%s",ss); s[i] = string(ss);
int len = strlen(ss);
for (int i = 0; i < len; i++) ins(ss[i] - ‘a‘);
}
int u;
for (int i = 1; i <= n; i++){
u = 1;
for (int j = 0; j < s[i].length(); j++){
u = ch[u][s[i][j] - ‘a‘];
for (int p = u; p && cur[p] != i; p = pre[p])
sz[p]++,cur[p] = i;
}
}
tsort();
sz[1] = 0;
for (int i = 1; i <= cnt; i++)
u = a[i],f[u] = f[pre[u]] + (sz[u] >= k ? step[u] - step[pre[u]] : 0);
for (int i = 1; i <= n; i++){
ans = 0; u = 1;
for (int j = 0; j < s[i].length(); j++){
u = ch[u][s[i][j] - ‘a‘];
ans += f[u];
}
printf("%lld ",ans);
}
return 0;
}
BZOJ3473 字符串 【廣義後綴自動機】