标签:des style blog color os io java ar strong
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 3580 Accepted Submission(s): 1037
ans是包含模式串的串个数,用总个数减去不包含的即可
总个数:f[i](小于等于i的所有字串)=f[i-1]*26+1,这个可以用矩阵乘法来实现
不包含的:如果单纯求走i步不包含,直接矩阵乘即可,现在要求1到i步的总和,在矩阵外加一维L+1,把a[i][L+1]都赋为1,即可实现求和
#include<cstdio> #include<cstdlib> #include<algorithm> #include<cmath> #include<cstring> #include<iostream> using namespace std; typedef unsigned long long ll; struct node{ node *fail; node *son[26]; bool pp; int num; node() { fail=NULL; for(int i=0;i<26;i++)son[i]=NULL; pp=true; num=0; } } *que[101]; node *ac[101]; node *root; ll a[101][101],b[101][101],c[101][101]; ll xzq,mus; char s[1011]; int len,tot,i,n,m; void Read() { char c; while(c=getchar(),c<‘a‘||c>‘z‘); s[++len]=c; while(c=getchar(),c>=‘a‘&&c<=‘z‘)s[++len]=c; } void add() { node *p; p=root; int i,t; for(i=1;i<=len;i++){ t=s[i]-‘a‘; if(p->son[t]==NULL){ p->son[t]=new node(); p->son[t]->num=++tot; ac[tot]=p->son[t]; } p=p->son[t]; } p->pp=false; } void bfs() { int l,r,i; node *p; node *q; que[l=r=1]=root; while(l<=r){ p=que[l]; for(i=0;i<=25;i++)if(p->son[i]!=NULL){ r++; que[r]=p->son[i]; if(p==root)p->son[i]->fail=root; else{ q=p->fail; while(q!=NULL){ if(q->son[i]!=NULL){ p->son[i]->fail=q->son[i]; break; } q=q->fail; } if(p->son[i]->fail==NULL)p->son[i]->fail=root; } } l++; } for(i=2;i<=r;i++)que[i]->pp&=que[i]->fail->pp; } void prepare() { node *p; node *q; int i,j; memset(a,0,sizeof(a)); for(i=1;i<=tot+1;i++)a[i][tot+1]=1; for(i=1;i<=tot;i++){ for(j=0;j<=25;j++){ p=ac[i]; if(p->son[j]!=NULL)p=p->son[j]; else{ q=p->fail; while(q!=NULL){ if(q->son[j]!=NULL){ p=q->son[j]; break; } q=q->fail; } if(q==NULL)p=root; } if(p->pp==true)a[i][p->num]++; } } } void mi(int x,int mx) { int i,j,k; if(x==1)return; mi(x/2,mx); for(i=1;i<=mx;i++){ for(j=1;j<=mx;j++){ c[i][j]=0; for(k=1;k<=mx;k++)c[i][j]=c[i][j]+a[i][k]*a[k][j]; } } for(i=1;i<=mx;i++) for(j=1;j<=mx;j++)a[i][j]=c[i][j]; if(x%2==1){ for(i=1;i<=mx;i++){ for(j=1;j<=mx;j++){ c[i][j]=0; for(k=1;k<=mx;k++)c[i][j]=c[i][j]+a[i][k]*b[k][j]; } } for(i=1;i<=mx;i++) for(j=1;j<=mx;j++)a[i][j]=c[i][j]; } } void work() { int i,j,li; mus=0; for(i=1;i<=tot+1;i++) for(j=1;j<=tot+1;j++)b[i][j]=a[i][j]; mi(m,tot+1); for(i=1;i<=tot+1;i++)mus+=a[1][i]; b[1][1]=26; b[1][2]=0; b[2][1]=1; b[2][2]=1; a[1][1]=26; a[1][2]=0; a[2][1]=1; a[2][2]=1; mi(m,2); xzq=a[1][1]+a[2][1]; xzq-=mus; } int main() { while(scanf("%d%d",&n,&m)!=EOF){ tot=0; root=new node(); root->num=++tot; ac[tot]=root; for(i=1;i<=n;i++){ len=0; Read(); add(); } bfs(); prepare(); work(); cout<<xzq<<endl; } }
标签:des style blog color os io java ar strong
原文地址:http://www.cnblogs.com/applejxt/p/3959637.html