bzoj1729：[Usaco2005 dec]Cow Patterns 牛的模式匹配（kmp+思维）

Problem

有一个 n n 个数的数列 A A ，其数字范围为 1 1 ~ k(k<=25) k ( k <= 25 ) 。
还有 m m 个数 B B 。
问从数列 n n 数列取出连续 m m 个数，排名与 B B 中排名一致的情况有多少种

Solution

我们考虑这个 B B ，将其变成所以可能的序列，然后和原串进行 kmp k m p
可能的序列可以用递归求解…
然而这样肯定复杂度大啊！

所以我们改进…（改进完和原来就没啥关系了…）
但本质还是改变模式串。
先来一波定义，定义 B B 中有 k k 个数

在求这个模式串的过程中，要是每位都去重新考虑就次数太多了…
我们将模式串进行一波离散化
那我们枚举模式串中的数字对应数列 A A 中为几，然后判断从哪些位开始的在模式串这个数字中成立。结果就是所以模式串每一个数字都模拟出来，且都包含的结果

说不清..上代码吧…

Code

#include 
#include 
#include 
using namespace std;
#define N 100010
#define inf 0x3f3f3f3f
int n,m,k,ans=0,nxt[N],f[N],s[N],t[N],a[N],b[N],c[N],maxx[N],maxy[N];
inline void KMP(int step,int x){
    for(int i=1;i<=n;i++) if(a[i]==x) s[i]=x;else s[i]=0;
    for(int i=1;i<=m;i++) if(b[i]==step) t[i]=x;else t[i]=0;
    nxt[1]=0; 
    for(int i=2,j=0;i<=m;i++){
        while(j && t[i]!=t[j+1]) j=nxt[j];
        if(t[i]==t[j+1]) j++;
        nxt[i]=j;
    }
    for(int i=1,j=0;i<=n;i++){
        while(j && (j==m || s[i]!=t[j+1])) j=nxt[j];
        if(s[i]==t[j+1]) j++;
        if(j==m) maxy[i-m+1]=x;
    }
}
int main(){
    while(scanf("%d%d%d",&n,&m,&k)>0){
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        memset(c,0,sizeof(c));
        for(int i=1;i<=m;i++) scanf("%d",&b[i]),c[b[i]]=1;
        memset(maxx,0,sizeof(maxx));
        for(int i=1;i<=k;i++){
            if(!c[i]) continue;
            for(int j=1;j<=n;j++) maxy[j]=inf;
            for(int j=1;j<=k;j++) 
                KMP(i,j);   //第i小的数j,看每位是否能通过这个数得到
            for(int j=1;j<=n;j++) if(maxx[j]>=maxy[j]) maxx[j]=inf;else maxx[j]=maxy[j];

        }
        for(int i=1;i<=n;i++) if(maxx[i]!=inf) ans++; 
        printf("%d\n",ans);
        for(int i=1;i<=n;i++) if(maxx[i]!=inf) printf("%d\n",i);
    }
    return 0;
}