hdu 3992 AC自动机上的高斯消元求期望

Crazy Typewriter

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 391    Accepted Submission(s): 109

Problem Description

There was a crazy typewriter before. When the writer is not very sober, it would type characters randomly, one per second, the possiblities of characters may differ.
The protagonist in this problem wants to tell acmers some secrets, of course, by the typewriter.

There had been several opportunities, but the protagonist let them sliped. Now, another opportunity came, the writer started a new paragraph. The protagonist found 
that he could set the possiblities of each character in happy astonishment. After the possiblities had been set, he wanted to know the exception of time at least the writer need to be mind-absent if any secret was typed out.

fewovigwnierfbiwfioeifaorfwarobahbgssjqmdowj

 

 

Input

There are several cases, no more than 15.
The first line of each case contains an integer n, no more than 15, indicating the number of secrets.
The second line contains 26 real numbers, indicating the set possibilities of 'a'-'z', respectively, the sum would be 1.0 .
Then n lines, each contains a secret, no longer than 15, which is made up by lowercase letters 'a'-'z'.

 

 

Output

A single line contains the expectation of time for each case， in seconds with six decimal, if the exception doesn't exist, output "Infinity"

 

 

Sample Input

2

0.5000 0.5000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000

ab

aa

 

 

Sample Output

3.000000

 

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <cmath>
using namespace std;

const int N=250;
const int SZ=26;
const double eps=1e-8;
double p[27],A[N][N];
bool inf[N];
struct AC
{
    int ch[N][26];
    int sz,val[N],fail[N];
    void init(){
        sz=1;memset(val,0,sizeof(val));
        memset(ch[0],0,sizeof(ch[0]));
    }
    void insert(char *s)
    {
        int rt=0;
        for (int i=0;s[i];i++){
            int c=s[i]-'a';
            if (ch[rt][c]==0){
                ch[rt][c]=sz;
                memset(ch[sz],0,sizeof(ch[sz]));
                sz++;
            }
            rt=ch[rt][c];
        }
        val[rt]=1;
    }
    void getfail()
    {
        queue<int>q;
        for (int i=0;i<SZ;i++)
        {
            int c=ch[0][i];
            if (c){ q.push(c);fail[c]=0; }
        }
        while (!q.empty())
        {
            int u=q.front();q.pop();
            for (int i=0;i<SZ;i++)
            {
                int c=ch[u][i];
                if (!c){
                    ch[u][i]=ch[fail[u]][i];
                }else 
                {
                    int v=fail[u];
                    q.push(c);
                    fail[c]=ch[v][i];
                    val[c]|=val[fail[c]];
                }
            }
        }
    }
}ac;

void build_matrix()
{
    int i,j;
    memset(A,0,sizeof(A));
    for(i=0;i<ac.sz;i++)
    {
        if(ac.val[i]){A[i][i]=1;A[i][ac.sz]=0;}
        else
        {
            for(j=0;j<SZ;j++){int v=ac.ch[i][j];A[i][v]+=p[j];}
            A[i][i]+=-1;A[i][ac.sz]=-1;
        }
    }
}

void gauss(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(fabs(A[r][i])<eps) continue;
            if(r!=i) for(j=0;j<=n;j++) swap(A[r][j],A[i][j]);
            for(k=0;k<n;k++) if(k!=i)
                for(j=n;j>=i;j--) A[k][j]-=A[k][i]/A[i][i]*A[i][j];
    }
    memset(inf,0,sizeof(inf));
    for(i=ac.sz-1;i>=0;i--){
        if(fabs(A[i][i])<eps&&fabs(A[i][ac.sz])>eps) inf[i]=1;
        for(j=i+1;j<ac.sz;j++)
            if(fabs(A[i][j])>eps&&inf[j]) inf[i]=1;
    }
    if(inf[0]) printf("Infinity\n");
    else printf("%.6lf\n",A[0][ac.sz]/A[0][0]+eps);
}

int main()
{
    //freopen("b.txt","w",stdout);
    int n,i,j;char s[20];
    while(~scanf("%d",&n))
    {
        ac.init();
        for(i=0;i<SZ;i++) scanf("%lf",p+i);
        for(i=0;i<n;i++)
        {
            scanf("%s",s);ac.insert(s);
        }
        ac.getfail();
        build_matrix();
        gauss(ac.sz);
    }
    return 0;
}