JSOI2007 文本生成器

题目链接:戳我

非常不好意思,因为想要排版,所以今天先只把代码贴出来,题解明天补。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<queue>
#define MAXN 6010
#define mod 10007
using namespace std;
int n,m,tot;
int dp[MAXN][MAXN];
char ss[MAXN];
struct Node{int ch[26],fail,end;}t[MAXN<<1];
inline int fpow(int x,int y)
{
    int cur_ans=1;
    while(y)
    {
        if(y&1) cur_ans=1ll*cur_ans*x%mod;
        x=1ll*x*x%mod;
        y>>=1;
    }
    return cur_ans;
}
inline void build(char *s)
{
    int len=strlen(s+1),x=0;
    for(int i=1;i<=len;i++)
    {
        if(!t[x].ch[s[i]-'A']) t[x].ch[s[i]-'A']=++tot;
        x=t[x].ch[s[i]-'A'];
    }
    t[x].end=1;
}
inline void get_fail()
{
    queue<int>q;
    int x=0;
    for(int i=0;i<26;i++)
        if(t[x].ch[i])
            t[t[x].ch[i]].fail=0,q.push(t[x].ch[i]);
    while(!q.empty())
    {
        int u=q.front();q.pop();
        t[u].end|=t[t[u].fail].end;
        for(int i=0;i<26;i++)
        {
            if(t[u].ch[i]) t[t[u].ch[i]].fail=t[t[u].fail].ch[i],q.push(t[u].ch[i]);
            else t[u].ch[i]=t[t[u].fail].ch[i];
        }
    }
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("ce.in","r",stdin);
    #endif
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",ss+1);
        build(ss);
    }
    get_fail();
    dp[0][0]=1;
    int ans=fpow(26,m)%mod,cur_ans=0;
    for(int i=1;i<=m;i++)
        for(int j=0;j<=tot;j++)
            for(int k=0;k<26;k++)
            {
                // cout<<t[j].ch[k]<<endl;
                if(t[t[j].ch[k]].end==0)
                {
                    dp[t[j].ch[k]][i]=(dp[t[j].ch[k]][i]+dp[j][i-1])%mod;
                    // printf("%d\n",dp[t[j].ch[k]][i]);
                }
            }
    for(int i=0;i<=tot;i++) cur_ans=(cur_ans+dp[i][m])%mod;
    // printf("cur_ans=%d\n",cur_ans);
    printf("%d\n",(ans+mod-cur_ans)%mod);
    return 0;
}
posted @ 2019-05-17 23:52  风浔凌  阅读(185)  评论(0编辑  收藏  举报