HDU 5955 AC自动机+高斯消元

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<queue>
using namespace std;
const int SIGMA_SIZE=6;
const int MAXNODE=100+20;
const int maxn=12;
int idx[10];
void init()
{
for(int i=0;i<6;i++){
idx[i+1]=i;
}
}
void Gauss(double A[MAXNODE][MAXNODE],int n)
{
int i,j,k,r;
for(i=0;i<n;i++){
r=i;
for(j=i+1;j<n;j++)
if(fabs(A[j][i])>fabs(A[r][i])) r=j;
if(r!=i)    for(j=0;j<=n;j++)   swap(A[r][j],A[i][j]);
for(k=i+1;k<n;k++){
double f=A[k][i]/A[i][i];
for(j=i;j<=n;j++)   A[k][j]-=f*A[i][j];
}
}
for(i=n-1;i>=0;i--){
for(j=i+1;j<n;j++){
A[i]
-=A[j]
*A[i][j];

}
A[i]
/=A[i][i];
}

}
struct ACAutomata{
int ch[MAXNODE][SIGMA_SIZE];
int f[MAXNODE];
int val[MAXNODE];
int last[MAXNODE];
int sz;
double A[MAXNODE][MAXNODE];
bool vis[MAXNODE];
void init()
{
memset(ch[0],0,sizeof(ch[0]));
sz=1;
}
void insert(int s[],int n,int v)
{
int u=0;
for(int i=0;i<n;i++){
int c=idx[s[i]];
if(!ch[u][c]){
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=0;
ch[u][c]=sz++;
}
u=ch[u][c];
}
val[u]=v;
}
void getFail()
{
queue<int> que;
int u=0;
f[0]=0;
for(int c=0;c<SIGMA_SIZE;c++){
u=ch[0][c];
if(u){
f[u]=0;last[u]=0;
que.push(u);
}
}
while(!que.empty()){
int r=que.front();que.pop();
for(int c=0;c<SIGMA_SIZE;c++){
u=ch[r][c];
if(!u){
ch[r][c]=ch[f[r]][c];
continue;
}
que.push(u);
int v=f[r];
while(v&&!ch[v][c]){
v=f[v];
}
f[u]=ch[v][c];
last[u]=val[f[u]]?f[u]:last[f[u]];
if(!val[u]){
val[u]=val[f[u]]?1:0;
}
}
}
}

double GetMatrix(int pos,int N)
{
memset(A,0,sizeof(A));
for(int u=0;u<sz;u++){
if(val[u]){
if(val[u]==pos){
A[u][sz]=1.0;
A[u][u]=1.0;
}
else{
A[u][u]=1.0;
A[u][sz]=0.0;
}
continue;
}
else{
A[u][u]=6.0;
for(int i=0;i<SIGMA_SIZE;i++){
//                  printf("%d %d\n",u,ch[u][i]);
A[u][ch[u][i]]-=1.0;
}
}
}
//      for(int i=0;i<sz;i++){
//          for(int j=0;j<=sz;j++){
//              printf("%0.2lf ",A[i][j]);
//          }
//          printf("\n");
//      }
Gauss(A,sz);
//      for(int i=0;i<sz;i++){
//          for(int j=0;j<=sz;j++){
//              printf("%0.2lf ",A[i][j]);
//          }
//          printf("\n");
//      }
return A[0][sz];
}
};
ACAutomata ac;
int str[maxn][maxn];
int T,N;
int len;
void solve();
int main()
{
init();
scanf("%d",&T);
while(T--){
solve();
}
return 0;
}
void solve()
{
ac.init();
scanf("%d %d",&N,&len);
for(int i=1;i<=N;i++){
for(int j=0;j<len;j++){
scanf("%d",&str[i][j]);
}
ac.insert(str[i],len,i);
}
ac.getFail();
for(int i=1;i<=N;i++){
double ans=ac.GetMatrix(i,N);
if(i!=1) printf(" ");
printf("%0.6lf",ans);
}
printf("\n");
}