KMP 题记

Poj-3461 Oulipo

• <模板题>求模式串T在主串S中出现的次数

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int Max_n=1e6+10;
const int Max_m=1e4+10;

int T;
char s[Max_n],t[Max_m];
int nex[Max_m];

void getNext(int tlen,int slen){
	int i=0,j=-1;
	nex[0]=-1;
	while(i<tlen){
		if(j==-1||t[i]==t[j]){
			i++;j++;
			nex[i]=j;
		}
		else j=nex[j];
	}
}

int solve(int tlen,int slen){
	int i=0,j=0,ans=0;
	while(i<slen){
		if(j==-1||s[i]==t[j]){
			i++;j++;
		}
		else j=nex[j];
		if(j==tlen){
			ans++;
			j=nex[j];
		}
	}
	return ans;
}

int main()
{
	scanf("%d",&T);
	while(T--){
		scanf("%s%s",t,s);
		int tlen=strlen(t);
		int slen=strlen(s);
		getNext(tlen,slen);
		printf("%d\n",solve(tlen,slen)); 
	}
	return 0;
 } 

Poj-2757 Seek the Name, Seek the Fame

• next数组的简单应用

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int Max_m=1e6+10;

char t[Max_m];
int nex[Max_m];

void getNext(int tlen){
	int i=0,j=-1;
	nex[0]=-1;
	while(i<tlen){
		if(j==-1||t[i]==t[j]){
			i++;j++;
			nex[i]=j;
		}
		else j=nex[j];
	}
}

void Print(int i){
	if(nex[i]==0){
		printf("%d",i);
		return;
	}
	Print(nex[i]);
	printf(" %d",i);
}

int main()
{
	while(~scanf("%s",t)){
		int tlen=strlen(t);
		getNext(tlen); 
		Print(tlen);
		putchar('\n'); 
	}
	return 0;
 } 

Poj-2406 Power Strings

• 串长为n，如果n%（n-next[n]）==0,则存在重复连续子串，长度为n-next[n]，否则输出1。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int Max_n=1e6+10;

char s[Max_n];
int nex[Max_n];

void getNext(int len){
	int i=0,j=-1;
	nex[i]=-1;
	while(i<len){
		if(j==-1||s[i]==s[j]){
			i++;j++;
			nex[i]=j;
		}
		else j=nex[j];
	}
}

int main()
{
	while(~scanf("%s",s)&&s[0]!='.'){
		int len=strlen(s);
		getNext(len);
		if(len%(len-nex[len])==0)printf("%d\n",len/(len-nex[len]));
		else printf("1\n");
	}
	return 0;
 } 

Poj-1916 Period

• 和上题一样，求这个字符串到第i个字符为止的循环节的次数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int Max_n=1e6+10;

int T=0,n;
char s[Max_n];
int nex[Max_n];

void getNext(int len){
	int i=0,j=-1;
	nex[0]=-1;
	while(i<len){
		if(j==-1||s[i]==s[j]){
			i++;j++;
			nex[i]=j;
		}
		else j=nex[j];
	}
}

int main()
{
	while(~scanf("%d",&n)&&n){
		scanf("%s",s);
		printf("Test case #%d\n",++T);
		int len=strlen(s);
		getNext(len);
		for(int i=2;i<=len;i++){
			if(nex[i]!=0&&i%(i-nex[i])==0)
				printf("%d %d\n",i,i/(i-nex[i]));
		}
		putchar('\n');
	}
	return 0;
}

51Nod-1277 字符串中的最大值

• KMP+DP求解。我们可以发现，前缀的前缀一定在前缀中重复出现，这句话可能不好理解，举个例子， abababc 他的一个前缀ababab中的前缀为ab 那么我们就可以知道 num[ab]+=num[ababab],根据这个地推关系，我们可以找出dp的状态转移方程，同时利用kmp 的next数组就可以了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int Max_n=1e5+10;

int len;
char t[Max_n];
int nex[Max_n];
ll ans[Max_n];

void getNext(){
	int i=0,j=-1;
	nex[0]=-1;
	while(i<len){
		if(j==-1||t[i]==t[j]){
			i++;j++;
			nex[i]=j;
		}
		else j=nex[j];
	}
}

int main()
{
	scanf("%s",t);
	len=strlen(t);
	getNext();
	memset(ans,0,sizeof(ans));
	ll Max=0;
	for(int i=len;i>=1;i--){
		ans[i]+=1;
		ans[nex[i]]+=ans[i];
		Max=max(ans[i]*i,Max);
	}
	printf("%lld\n",Max);
	return 0;
}