码迷,mamicode.com
首页 > 其他好文 > 详细

【BZOJ】3670 [Noi2014]动物园

时间:2017-03-27 14:22:35      阅读:190      评论:0      收藏:0      [点我收藏+]

标签:href   算法   ref   article   std   max   匹配   pre   class   

【算法】KMP

【题解】题目要求的是数量而非长度……QAQ

BZOJ 3670 NOI2014 动物园 KMP算法

跑KMP预匹配实际上是建一颗KMP树,跑失配边其实就是跑向父亲,一个节点的儿子失配后就会跑向它。

理解了KMP树后就很简单了,题目要求的是满足要求(长度<一半)的前后缀相等的数量,其实等价于节点i的祖先的数量。

所以用cnt[i]统计深度,在跑KMP的时候顺便跑出来统计答案即可。

KMP的时候直接while往前跑直到满足要求,由KMP中j每次至多+1的性质可知总效率是O(n)。

由于祖先的cnt一定计算过,不用考虑后效性,所以两个过程可以合并一起写。

技术分享
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxL=1000010,MOD=1000000007;
long long ans=1;
int tt,p[maxL],cnt[maxL];
char s[maxL];
int main()
{
    scanf("%d",&tt);
    while(tt--)
     {
         scanf("%s",s+1);
         int n=strlen(s+1);
         int j=0,k=0;cnt[1]=1;p[1]=0;ans=1;
         for(int i=2;i<=n;i++)
          {
             while(j>0&&s[i]!=s[j+1])j=p[j];
             while(k>0&&s[i]!=s[k+1])k=p[k];
             if(s[i]==s[j+1])j++,k++;
             p[i]=j;
             cnt[i]=cnt[j]+1;
             if(k*2>i)k=p[k];
             ans=(ans*(cnt[k]+1))%MOD;
          }
         printf("%lld\n",ans);
     }
    return 0;
}
View Code

【BZOJ】3670 [Noi2014]动物园

标签:href   算法   ref   article   std   max   匹配   pre   class   

原文地址:http://www.cnblogs.com/onioncyc/p/6626413.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!