听说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;
}