BZOJ1444: [Jsoi2009]有趣的游戏（Trie图，矩乘）

Description

Input

注意 是0<=P, n , l, m≤ 10.

Output

Sample Input

input 1
3 2 2
1 2
1 2
AB
BA
AA
input 2
3 4 2
1 2
1 2
AABA
ABAA
BAAA

Sample Output

output 1
0.25
0.50
0.25

output 2
0.31
0.33
0.37

解题思路：

可以理解为Trie上倍增佛洛依德做矩乘累和。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
struct trnt{
    int ch[26];
    int fl;
    bool fin;
}tr[111];
class queue{
    public:
        queue(void){h=1,t=0;return ;}
        int nxt(int x){if(x+1>=999)return 1;return x+1;}
        void push(int x){t=nxt(t);line[t]=x;return ;}
        void pop(void){h=nxt(h);return ;}
        bool empty(void){return nxt(t)==h;}
        int front(void){return line[h];}
    private:
        int h,t,line[1000];
}Q;
int siz;
int n,m,l;
int ansp[111];
char tmp[111];
double lst[111];
double ans[111];
double martix[111][111];
double temp[111][111];
double pre[30];
void Insert(char *a,int &plc)
{
    int root=0;
    int len=l;
    for(int i=1;i<=len;i++)
    {
        int c=a[i]-'A';
        if(!tr[root].ch[c])
            tr[root].ch[c]=++siz;
        root=tr[root].ch[c];
    }
    tr[root].fin=true;
    plc=root;
    return ;
}
void Build(void)
{
    int root=0;
    for(int i=0;i<m;i++)
        if(tr[root].ch[i])
            Q.push(tr[root].ch[i]);
    while(!Q.empty())
    {
        root=Q.front();
        Q.pop();
        for(int i=0;i<m;i++)
        {
            if(tr[root].ch[i])
            {
                tr[tr[root].ch[i]].fl=tr[tr[root].fl].ch[i];
                Q.push(tr[root].ch[i]);
            }else
                tr[root].ch[i]=tr[tr[root].fl].ch[i];
        }
    }
    return ;
}
int main()
{
    scanf("%d%d%d",&n,&l,&m);
    for(int i=0;i<m;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        pre[i]=(double)(a)/(double)(b);
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%s",tmp+1);
        Insert(tmp,ansp[i]);
    }
    Build();
    for(int i=0;i<=siz;i++)
    {
        if(tr[i].fin)
        {
            martix[i][i]=1.00;
            continue;
        }
        for(int j=0;j<m;j++)
            martix[i][tr[i].ch[j]]+=pre[j];
        
    }
    int t=70;
    /*for(int i=0;i<=siz;i++)
    {
        for(int j=0;j<=siz;j++)
            printf("%.2lf ",martix[i][j]);
        puts("");
    }*/
    while(t--)
    {
        for(int i=0;i<=siz;i++)
        {
            for(int j=0;j<=siz;j++)
            {
                temp[i][j]=0;
                for(int k=0;k<=siz;k++)
                {
                    temp[i][j]+=martix[i][k]*martix[k][j];
                }
            }
        }
        for(int i=0;i<=siz;i++)
            for(int j=0;j<=siz;j++)
                martix[i][j]=temp[i][j];
    }
    for(int i=1;i<=n;i++)
        printf("%.2lf
",martix[0][ansp[i]]);
    return 0;
}