POJ 3167 Cow Patterns （KMP+前缀和）

题意：给你两串数字，长度分别为n和m，数字大小在[1,25]。当后一串数字每个数字的排名位置与前一串数字（任一长度为m的子串）每个数字的排名位置一致时就完全匹配，最后求哪些位置是完全匹配的。

例如：1 4 2 5 3 6 与 1 3 2 4  答案就是：1 3（第一串数字的第一个位置开始与第三个位置开始）

挺难想的一个题，我们需要使用dp的思想加前缀和进行KMP匹配

首先，两串数字串完全匹配就可以想到KMP，这样我们就只需要解决一个问题：两个数字怎样才算是“相等”（这儿并不是值一样就“相等”）。我们可以这样想，当两串数字对应位置：前面比此位置数字小的，与此位置数字相等的个数都一样，这样就一定匹配，因为这样每个数字排名位置一定相等（注意第一个数字串是子串）。

我们可以注意数字范围很小，所以可以使用前缀和记录：每个位置前面每种数字不大于这个数字大小的个数。这样求Next数组和匹配时直接使用前缀和来处理就好。

#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<string>
#include<cstdio>
#include<cstring>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define eps 1E-8
/*注意可能会有输出-0.000*/
#define Sgn(x) (x<-eps? -1 :x<eps? 0:1)//x为两个浮点数差的比较,注意返回整型
#define Cvs(x) (x > 0.0 ? x+eps : x-eps)//浮点数转化
#define zero(x) (((x)>0?(x):-(x))<eps)//判断是否等于0
#define mul(a,b) (a<<b)
#define dir(a,b) (a>>b)
typedef long long ll;
typedef unsigned long long ull;
const int Inf=<<;
const double Pi=acos(-1.0);
const int Mod=1e9+;
const int Max=;
int Next[Max],k;
struct node//前i个字符中大于等于某种数字的总个数
{
    int num[Max][];
};
void Init(int n,int *str,node &dp)//预处理dp
{
    memset(dp.num[],,sizeof(dp.num[]));
    for(int i=;i<n;++i)
    {
        dp.num[i][]=;
        for(int j=;j<=k;j++)
        {
            dp.num[i][j]=dp.num[i-][j];
            if(j>=str[i-])
                dp.num[i][j]=dp.num[i-][j]+;
        }
    }
    return;
}
int Jud(int i,int j,node &dpm,node &dpp,int *strm,int *strp)//模式串与匹配串是否匹配
{
    int lim=j-i;
    if(dpm.num[i][strm[i]]==dpp.num[j][strp[j]]-dpp.num[lim][strp[j]]&&dpm.num[i][strm[i]-]==dpp.num[j][strp[j]-]-dpp.num[lim][strp[j]-])//这儿使用前缀和计算时要注意固定不大于某个值
    return ;
    return ;
}
void GetNext(int m,int *str,node &dp)//求next数组
{
    Next[]=-;
    int i=-,j=;
    while(j<m)
    {
        if(i==-||Jud(i,j,dp,dp,str,str))//当此字符前面小于等于其的都相等时就匹配
        {
            ++i,++j;
            Next[j]=i;
        }
        else
            i=Next[i];
    }
    return;
}
int strm[Max],strp[Max];
node dpm,dpp;
int ans[Max],cnt;
int Kmp(int n,int m)//标准模式匹配修改
{
    int res=;
    int i=,j=;
    while(j<n)
    {
        if(i==-||Jud(i,j,dpm,dpp,strm,strp))
        {
            ++i,++j;
        }
        else
        i=Next[i];
        if(i==m)//匹配成功
        {
            ans[cnt++]=j-i+;
            i=Next[i];
            res++;
        }
    }
    return res;
}
int main()
{
    int n,m;
    while(~scanf("%d %d %d",&n,&m,&k))
    {
        cnt=;
        for(int i=;i<n;++i)
        scanf("%d",&strp[i]);
        for(int i=;i<m;++i)
        scanf("%d",&strm[i]);
        Init(m,strm,dpm);
        GetNext(m,strm,dpm);
        Init(n,strp,dpp);
        int len=Kmp(n,m);
        printf("%d\n",len);
        for(int i=;i<len;++i)
            printf("%d\n",ans[i]);
    }
    return ;
}