先对原串做一遍kmp,然后对每个结点,建立状态转移矩阵,然后做高斯消元即可得到结果。注意这里不可以用实数运算,否则会因掉精度而WA,所以需要避免实数运算。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
const int maxn=20;
typedef long long matrix[maxn][maxn];
int n,m;
char s[maxn];
int f[maxn];
void getfail(char *p)
{
m=strlen(p);
f[0]=0;f[1]=0;
for(int i=1;i<m;i++)
{
int j=f[i];
while(j && p[i]!=p[j]) j=f[j];
f[i+1]=p[i]==p[j]?j+1:0;
}
}
void gauss(matrix A,int n)
{
int i,j,k,r;
for(i=0;i<n;i++)
{
for(j=n;j>=i;j--)
for(k=i+1;k<n;k++)
A[k][j]-=A[k][i]/A[i][i]*A[i][j];
}
for(i=n-1;i>=0;i--)
{
for(j=i+1;j<n;j++)
A[i][n]-=A[j][n]*A[i][j];
A[i][n]/=A[i][i];
}
}
int main()
{
//freopen("in.txt","r",stdin);
int T,kase=1;
scanf("%d",&T);
while(T--)
{
if(kase>1) puts("");
scanf("%d%s",&n,s);
getfail(s);
matrix a;
memset(a,0,sizeof(a));
for(int i=0;i<m;i++)
{
a[i][i]=a[i][m]=n;
for(int j=0;j<n;j++)
{
int t=i;
while(t && s[t]!='A'+j) t=f[t];
if(s[t]=='A'+j)
{
if(t+1<m) a[i][t+1]-=1;
}
else a[i][t]-=1;
}
}
// for(int i=0;i<m;i++)
// {
// for(int j=0;j<=m;j++)
// cout<<a[i][j]<<" ";
// cout<<endl;
// }
gauss(a,m);
printf("Case %d:\n",kase++);
printf("%lld\n",a[0][m]);
}
return 0;
}