【KMP】【poj3167】【bzoj1729】Cow Patterns

• 题意
  给定一个顺序序列表示奶牛的排列顺序，定义串a=b当且仅当a的每位的排名=b的每位的排名。
• 题解
  预处理出前i位中j出现的次数，由于编号很小，处理出类似前缀和，暴力查询排名即可。已知排名之后就直接kmp。
• 代码

#include <iostream>
#include <cstdio>
#include <cstring>
#define MAXN 100001
using namespace std;
int n,m,s;
int s1[MAXN],s2[MAXN];
int p1[MAXN][30],p2[MAXN][30];
void init(){
    for(int i=1;i<=n;i++){
        for(int j=1;j<=s;j++)
            p1[i][j]=p1[i-1][j];
        p1[i][s1[i]]++;
    }
    for(int i=1;i<=m;i++){
        for(int j=1;j<=s;j++)
            p2[i][j]=p2[i-1][j];
        p2[i][s2[i]]++;
    }
}
int nxt[MAXN];
void getnxt(){
    int fix=0;
    int i=1;
    while(i<=m){
        int u1=0,u2=0;
        int a1=0,a2=0;
        for(int j=1;j<s2[i];j++)//i fix+1
            u1+=p2[i][j]-p2[i-fix][j];
        a1=p2[i][s2[i]]-p2[i-fix][s2[i]];
        for(int j=1;j<s2[fix];j++)
            u2+=p2[fix][j];
        a2=p2[fix][s2[fix]];
        if(fix==0||(a1==a2&&u1==u2)) i++,fix++,nxt[i]=fix;
        else fix=nxt[fix];
    }
}
int ans[MAXN],tot;
void kmp(){
    int i=1,j=1;
    while(i<=n){
        int u1=0,u2=0;
        int a1=0,a2=0;
        for(int k=1;k<s1[i];k++)
            u1+=p1[i][k]-p1[i-j][k];
        a1=p1[i][s1[i]]-p1[i-j][s1[i]];
        for(int k=1;k<s2[j];k++)
            u2+=p2[j][k];
        a2=p2[j][s2[j]];
        if(!j||(a1==a2&&u1==u2)) j++,i++;
        else j=nxt[j];
        if(j==m+1){
            ans[++tot]=i-m;
            j=nxt[j];
        }
    }
}
int main(){
    while(~scanf("%d%d%d",&n,&m,&s)){
        memset(nxt,0,sizeof nxt);
        memset(p1,0,sizeof p1);
        memset(p2,0,sizeof p2);
        tot=0;
        for(int i=1;i<=n;i++) scanf("%d",&s1[i]);
        for(int i=1;i<=m;i++) scanf("%d",&s2[i]);
        init();
        getnxt();
        kmp();
        printf("%d\n",tot);
        for(int i=1;i<=tot;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}