字符串 KMP HDU 4749

#include <stdio.h>
#include <string.h>
/*
10 5 10
2 4 2 4 2 4 2 4 2 4
1 2 1 2 1
题目大意：给一个N,M,K，序列a[],b[]，在a[]中找出，和b[]相对大小一样的子序列 
求序列的个数（序列不能重复）  
思路：KMP，找到满足条件，跳到i+m处 继续往下搜 
主要难点在于，如何判断满足条件。
我们只要比较i之前a中a[i]的个数和B中B[i]的个数是否相等，
以及a中小于a[i]的个数和B中小于B[i]的数字个数是否相等。
*/
const int N = 100005;
int n,m,k;
int a[N],b[N],next[N];
int f[30][N],g[30][N];
int judge(int j,int i)
{
	int eq1=0,lt1=0,eq2=0,lt2=0;
	for(int x=1;x<=k;x++)
	{
		if(x<a[i]){
			lt1+= f[x][i] - f[x][i-j];
		}
		else if( x==a[i]){
			eq1+= f[x][i] - f[x][i-j];
		}
		
		//当前b[] 位置j  
		if(x<b[j]){
			lt2+= g[x][j];
		}
		else if( x==b[j]){
			eq2+= g[x][j];
		}
	}
	return (lt1==lt2 && eq1==eq2);
} 
void getNext()
{
	int j=0;next[1]=0;
	for(int i=2;i<=m;i++)
	{
		while(j>0 && !judge(j+1,i))
			j=next[j];
		if(judge(j+1,i)) j++;
		next[i]=j;
	}
}
int kmp()
{
	int ans=0;
	int j=0;
	for(int i=1;i<=n;i++)
	{
		while(j>0 && !judge(j+1,i))
			j=next[j];
		if(judge(j+1,i))
			j++;
		if(j==m)
		{
			ans++; j=0;//继续往下找 
		}
	}
	return ans;
}
int main()
{
	while(scanf("%d%d%d",&n,&m,&k)!=EOF )
	{
		memset(f,0,sizeof(f));
		memset(g,0,sizeof(g)); 
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			for(int j=1;j<=k;j++)
				f[j][i]= f[j][i-1];
			f[a[i]][i]++;
		}	
		for(int i=1;i<=m;i++)
		{
			scanf("%d",&b[i]);
			for(int j=1;j<=k;j++)
				g[j][i]= g[j][i-1];
			g[b[i]][i]++;
		}	
		getNext();
		printf("%d\n",kmp());
	}
	return 0;
}