注册 登录  
 加关注
   显示下一条  |  关闭
温馨提示!由于新浪微博认证机制调整,您的新浪微博帐号绑定已过期,请重新绑定!立即重新绑定新浪微博》  |  关闭

ydc的博客

 
 
 

日志

 
 

bzoj1009  

2013-01-23 12:58:46|  分类: bzoj |  标签: |举报 |字号 订阅

  下载LOFTER 我的照片书  |

GT考试

听说是KMP+矩乘吧?

我写了个AC自动机+矩乘……KMP似乎不是很熟啊

任意一个串,就对应AC自动机的一个节点,而AC自动机的元素个数又少的可怜,就可以拿AC自动机的节点为状态DP然后求出递推式再用矩阵优化就行了。

AC自动机刚开始写错了……我还有什么话可说呢?

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#define MAXN 30
#define SIZE 15
using namespace std;
int n,m,mod,tot;
char a[MAXN];
queue<int>q;
struct AC
{
    int son[15],next,last;
    bool mark;
}tree[MAXN];
struct Matrix
{
    int digit[MAXN][MAXN],n,m;
    friend Matrix operator * (const Matrix &a,const Matrix &b)
    {
        Matrix c;
        memset(c.digit,0,sizeof(c.digit));
        c.n=a.n,c.m=b.m;
        for(int i=1;i<=a.n;i++)
            for(int j=1;j<=a.m;j++)
                for(int k=1;k<=b.m;k++)
                    c.digit[i][k]=(c.digit[i][k]+a.digit[i][j]*b.digit[j][k])%mod;
        return c;
    }
}ans,p;
void Insert(char a[])
{
    int root=0;
    for(int i=0;a[i];i++)
    {
        if(tree[root].son[a[i]-'0']==0)  tree[root].son[a[i]-'0']=++tot;
        root=tree[root].son[a[i]-'0'];
    }
    tree[root].mark=true;
}
void read()
{
    scanf("%d %d %d %s",&n,&m,&mod,a);
    Insert(a);
}
void Bfs()
{
    for(int i=0;i<=9;i++)
        if(tree[0].son[i])
            q.push(tree[0].son[i]);
    int p;
    while(q.empty()==false)
    {
        p=q.front(),q.pop();
        for(int i=0;i<=9;i++)
            if(tree[p].son[i])
            {
                int u=tree[p].son[i],v=tree[p].next;
                for(;v&&tree[v].son[i]==0;v=tree[v].next);
                tree[u].next=v=tree[v].son[i],q.push(u),tree[u].last=tree[v].mark==true?v:tree[v].last;
            }
    }
}
void make_Matrix()
{
    p.n=tot+1,p.m=tot+1;
    for(int i=0;i<=tot;i++)
        if(tree[i].mark==false)
            for(int j=0;j<=9;j++)
            {
                int k=i;
                for(;k&&tree[k].son[j]==0;k=tree[k].next);
                k=tree[k].son[j];
                if(tree[k].mark||tree[k].last)  continue;
                p.digit[i+1][k+1]++;
            }
    ans.n=1,ans.m=tot+1,ans.digit[1][1]=1;
}
Matrix power(int n,Matrix p)
{
    Matrix ans;
    ans.n=p.n,ans.m=p.m;
    for(int i=1;i<=p.n;i++)
        for(int j=1;j<=p.m;j++)
            ans.digit[i][j]=(i==j);
    while(n)
    {
        if(n&1)  ans=ans*p;
        p=p*p,n>>=1;
    }
    return ans;
}
int solve()
{
    int total=0;
    for(int i=1;i<=ans.m;i++)
        total=(total+ans.digit[1][i])%mod;
    return total;
}
int main()
{
    read();
    Bfs();
    make_Matrix();
    p=power(n,p),ans=ans*p;
    printf("%d\n",solve());
    return 0;
}
  评论这张
 
阅读(545)| 评论(0)
推荐 转载

历史上的今天

在LOFTER的更多文章

评论

<#--最新日志,群博日志--> <#--推荐日志--> <#--引用记录--> <#--博主推荐--> <#--随机阅读--> <#--首页推荐--> <#--历史上的今天--> <#--被推荐日志--> <#--上一篇,下一篇--> <#-- 热度 --> <#-- 网易新闻广告 --> <#--右边模块结构--> <#--评论模块结构--> <#--引用模块结构--> <#--博主发起的投票-->
 
 
 
 
 
 
 
 
 
 
 
 
 
 

页脚

网易公司版权所有 ©1997-2017