BZOJ 3670: [Noi2014]动物园&&51nod 1277 字符串中的最大值

最新推荐文章于 2020-01-20 20:03:41 发布

sillyf

最新推荐文章于 2020-01-20 20:03:41 发布

阅读量250

点赞数

分类专栏： kmp 文章标签： kmp

本文链接：https://blog.csdn.net/qq_35878547/article/details/76528096

版权

kmp 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

听说kmp的next数组可以用来建树,而且很有用

BZOJ 3670: [Noi2014]动物园

脑补一棵next树,将前缀长度当做编号,每一个i向next[i]连边,next[i]是i的父亲,
那么num[i]就是编号为i的节点的祖先中编号最大的节点的深度
但是实现并不一定要真的建树

#include<cstdio>
#include<cstring>
#define N 1000005
#define LL long long
#define p 1000000007
using namespace std;

int t,n;
int next[N],deep[N];
char s[N];

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s",s+1);n=strlen(s+1);
        int now=0;
        next[1]=0,deep[1]=1;
        for(int i=2;i<=n;++i)
        {
            while(now&&s[now+1]!=s[i]) now=next[now];
            if(s[now+1]==s[i]) ++now;
            next[i]=now;
            deep[i]=deep[now]+1;
        }
        LL ans=1;
        now=0;
        for(int i=2;i<=n;++i)
        {
            while(now&&s[now+1]!=s[i]) now=next[now];
            if(s[now+1]==s[i]) ++now;
            while(now*2>i) now=next[now];
            ans=ans*(deep[now]+1)%p;
        }
        printf("%lld\n",ans);
    }
    return 0;
}

——————————————————————————————————————————————————————–

51nod 1277 字符串中的最大值

用next数组建树,用前缀长度做节点编号,计算答案时直接在树上找
这个前缀出现的次数就是它的子节点个数加上它本身
还是可以不用建树

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define LL long long
using namespace std;

int n,now;
int next[N],cnt[N];
char s[N];

int main()
{
    scanf("%s",s+1);n=strlen(s+1);
    now=0;
    for(int i=2;i<=n;++i)
    {
        while(now&&s[now+1]!=s[i]) now=next[now];
        if(s[now+1]==s[i]) ++now;
        next[i]=now;
        cnt[now]++;
    }
    LL ans=0;
    for(int i=n;i;--i)
    {
        ans=max(ans,1ll*i*(cnt[i]+1));
        cnt[next[i]]+=cnt[i];
    }
    printf("%lld\n",ans);
    return 0;
}