标签:insert lin div san span void space har pre
题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=3172
对这$n$个串建立AC自动机,求每个串的出现次数。
我们在建立时对每一个经过的节点sum++,然后在fail树上从叶子节点不断向上更新,因为父亲节点代表的是孩子节点的一个后缀,所以孩子出现过的父亲一定出现过,而fail指针的不遗漏性保证了答案不会丢失。
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 using namespace std; 5 int N,ch[1000010][26],fail[1000010],cnt=0; 6 int sum[1000010],pos[1000010]; 7 char tmp[1000010]; 8 int Insert(char *s){ 9 int now=0; 10 for(int i=0;s[i];i++){ 11 int idx=s[i]-‘a‘; 12 if(!ch[now][idx]) ch[now][idx]=++cnt; 13 now=ch[now][idx]; 14 sum[now]++; 15 } 16 return now; 17 } 18 int q[1000010]; 19 void Setfail(){ 20 q[1]=0; 21 int head=1,tail=1; 22 while(head<=tail){ 23 int now=q[head]; 24 for(int i=0;i<26;i++){ 25 if(ch[now][i]){ 26 q[++tail]=ch[now][i]; 27 fail[ch[now][i]]=now?ch[fail[now]][i]:0; 28 } 29 else ch[now][i]=ch[fail[now]][i]; 30 } 31 head++; 32 } 33 for(int i=tail;i>=1;i--) sum[fail[q[i]]]+=sum[q[i]]; 34 } 35 int main(){ 36 scanf("%d",&N); 37 for(int i=1;i<=N;i++){ 38 scanf("%s",tmp); 39 pos[i]=Insert(tmp); 40 } 41 Setfail(); 42 for(int i=1;i<=N;i++) printf("%d\n",sum[pos[i]]); 43 return 0; 44 }
标签:insert lin div san span void space har pre
原文地址:http://www.cnblogs.com/halfrot/p/7635899.html