#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;
}
字符串 KMP HDU 4749
最新推荐文章于 2017-09-19 16:57:08 发布