CF432D Prefixes and Suffixes kmp+dp

原题:http://codeforces.com/contest/432/problem/D

题解:求所有前后缀相同的子串的个数。kmp中的nxt数组就是前后缀相等的最大长度,只要将j=nxt[nxt[i]]做到刚好大于0就行了。考虑dp有:f[i]表示1-i的字符串出现的个数,f[nxt[i]]+=f[i],所有应该从后往前扫描统计答案。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5+10;
char s[N];
int f[N],nxt[N],n,num,ans[N],len[N];
int main(){
//	freopen("cf432d.in","r",stdin);
	scanf("%s",s+1);n=strlen(s+1);
	nxt[0]=-1;
	for(int i=1,j;i<=n;i++){
		j=nxt[i-1];
		while(j!=-1 && s[i] != s[j+1]) j=nxt[j];
		nxt[i]=++j;		
	}
	for(int i=n;i>=1;i--){f[i]++;f[nxt[i]]+=f[i];}
	num=0;
	for(int i=n;i>=1;i=nxt[i]){ans[++num]=f[i];len[num]=i;}	
	printf("%d\n",num);
	for(int i=num;i>=1;i--) printf("%d %d\n",len[i],ans[i]);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_39689721/article/details/87298087
今日推荐