【AC自动机】【高斯消元】hdu5955 Guessing the Dice Roll

#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
#define N 110 
double B[N][N+1],A[N][N+1],x[N],b[N];
int ma[13],size;
void guass_jordan()
{
	int n=size;
    memcpy(B,A,sizeof(A));
    for(int i=1;i<=n;++i)
      B[i][n+1]=b[i];
    for(int i=1;i<=n;++i)//枚举:正在消除第i个未知数,之后第i个方程废掉,矩阵行、列-1
      {
        int pivot=i;
        for(int j=i+1;j<=n;++j)//枚举j:把正在处理的未知数的系数的绝对值最大的方程换到第i行
          if(fabs(B[j][i])>fabs(B[pivot][i]))
            pivot=j;
        swap(B[i],B[pivot]);
        //if(fabs(B[i][i])<EPS)
        //若所有(最大)的该未知数系数为0,说明少一个未知数,有无穷多解
        for(int j=i+1;j<=n+1;++j)
          B[i][j]/=B[i][i];
        for(int j=1;j<=n;++j)
          if(i!=j)//枚举所有的方程,从第j个方程中消去第i个未知数
            for(int k=i+1;k<=n+1;++k)//依次把第j个方程中的第k个未知数减去应减的数值
              B[j][k]-=B[j][i]*B[i][k];
      }
    for(int i=1;i<=n;++i) x[i]=B[i][n+1];
}
queue<int>q;
int n,L;
int child[110][6],fail[110];
bool tag[110];
void Insert(int S[],int id)
{
	int now=0;
	for(int i=1;i<=L;++i)
	  {
	  	if(!child[now][S[i]-1])
	  	  child[now][S[i]-1]=size++;
	  	now=child[now][S[i]-1];
	  }
	tag[now]=1;
	ma[id]=now;
}
void build()
{
	fail[0]=-1;
	q.push(0);
	while(!q.empty())
	  {
	  	int U=q.front(); q.pop();
	  	for(int i=0;i<6;++i)
	  	  if(child[U][i])
	  	    {
	  	      int V=fail[U];
	  	      while(V!=-1)
	  	        {
	  	          if(child[V][i])
	  	            {
	  	              fail[child[U][i]]=child[V][i];
	  	              break;
	  	            }
	  	          V=fail[V];
	  	        }
	  	      if(V==-1)
	  	        fail[child[U][i]]=0;
	  	      if(tag[fail[child[U][i]]])
                tag[child[U][i]]=1;
	  		  q.push(child[U][i]);
	  	    }
	  	  else if(U)
            child[U][i]=child[fail[U]][i];
	  }
}
void Init()
{
	memset(child,0,sizeof(child));
	memset(fail,0,sizeof(fail));
	memset(tag,0,sizeof(tag));
	memset(A,0,sizeof(A));
	memset(B,0,sizeof(B));
	memset(x,0,sizeof(x));
	memset(b,0,sizeof(b));
	size=1;
}
int T;
int main()
{
//	freopen("hdu5955.in","r",stdin);
	int s[13];
	scanf("%d",&T);
	for(;T;--T)
	  {
	  	Init();
	  	scanf("%d%d",&n,&L);
	  	for(int i=1;i<=n;++i)
	  	  {
	  	  	for(int j=1;j<=L;++j)
	  	  	  scanf("%d",&s[j]);
	  	  	Insert(s,i);
	  	  }
	  	build();
	  	for(int i=0;i<size;++i)
	  	  {
	  	  	if(!tag[i])
	  	  	  for(int j=0;j<6;++j)
	  	  	    A[child[i][j]+1][i+1]+=1.0/6.0;
	  	  	A[i+1][i+1]-=1.0;
	  	  }
	  	b[1]=-1.0;
	  	guass_jordan();
	  	for(int i=1;i<n;++i)
	  	  printf("%.6lf ",x[ma[i]+1]);
	  	printf("%.6lf\n",x[ma[n]+1]);
	  }
	return 0;
}