【洛谷 P4070】 [SDOI2016]生成魔咒(后缀自动机)
题目链接
建出\(SAM\)后,不同子串个数就是\(\sum len(i)-len(fa(i))\)
因为\(SAM\)在线的,所以每加入一个字符就能直接加上其贡献,于是这道题就没了。
因为\(x\)有点大,所以把\(ch\)改成\(map\)就行。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
const int MAXN = 1000010;
struct SAM{
map <int, int> ch;
int len, fa;
}sam[MAXN << 1];
int las = 1, cnt = 1;
long long ans;
inline void add(int c){
int p = las; int np = las = ++cnt;
sam[np].len = sam[p].len + 1;
for(; p && !sam[p].ch[c]; p = sam[p].fa) sam[p].ch[c] = np;
if(!p) sam[np].fa = 1;
else{
int q = sam[p].ch[c];
if(sam[q].len == sam[p].len + 1) sam[np].fa = q;
else{
int nq = ++cnt; sam[nq] = sam[q];
sam[nq].len = sam[p].len + 1;
sam[q].fa = sam[np].fa = nq;
for(; p && sam[p].ch[c] == q; p = sam[p].fa) sam[p].ch[c] = nq;
}
}
ans += sam[np].len - sam[sam[np].fa].len;
}
int a, n;
int main(){
scanf("%d", &n);
for(int i = 1; i <= n; ++i){
scanf("%d", &a);
add(a);
printf("%lld\n", ans);
}
return 0;
}