NOIP 考前 KMP练习

BZOJ 1461 && BZOJ 1729

KMP+BIT

一看就是字符串匹配但是不同的是要按照每个字符的排名情况。

首先对于数字x的排名，那么要判断x前小于x的数的个数，和x前小于等于x的数的个数，这两个都相等才相等。因为会有数字会有重复的.

然后可以先预处理出字串的每个数字的排名，这样就不需要两个BIT了。最后按照KMP的思路做就可以了.

#include <cstdio>
#include <cstring>
const int MaxN=500100;
const int MaxS=10010;
int a[MaxN],b[MaxN],Rk1[MaxN],Rk2[MaxN],Ans[MaxN],tot,n,k,s,P[MaxN];
int c[MaxS];
inline int Lowbit(int x) {return x&-x;}
inline void Add(int x,int d)
{for (int i=x;i<=s;i+=Lowbit(i)) c[i]+=d;}
inline int Query(int x)
{
    int Ret=0;
    for (int i=x;i;i-=Lowbit(i)) Ret+=c[i];
    return Ret;
}
int main()
{
    scanf("%d%d%d",&n,&k,&s);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1;i<=k;i++)
    {
        scanf("%d",&b[i]); Add(b[i],1);
        Rk1[i]=Query(b[i]),Rk2[i]=Query(b[i]-1);
    }
    memset(c,0,sizeof(c)); int j=0;
    for (int i=2;i<=k;i++)
    {
        Add(b[i],1);
        while (j && ((Query(b[i])!=Rk1[j+1]) || (Query(b[i]-1)!=Rk2[j+1])))
        {
            for (int l=i-j;l<i-P[j];l++) Add(b[l],-1);
            j=P[j];
        }
        if ((Query(b[i])==Rk1[j+1]) && (Query(b[i]-1)==Rk2[j+1])) j++;
        P[i]=j;
    }
    memset(c,0,sizeof(c)); j=0;
    for (int i=1;i<=n;i++)
    {
        Add(a[i],1);
        while (j && (Query(a[i])!=Rk1[j+1] || Query(a[i]-1)!=Rk2[j+1])) 
        {
            for (int l=i-j;l<i-P[j];l++) Add(a[l],-1);
            j=P[j];
        }
        if (Query(a[i])==Rk1[j+1] && Query(a[i]-1)==Rk2[j+1]) j++;
        if (j==k)
        {
            Ans[++tot]=i-j+1;
            for (int l=i-j+1;l<=i-P[j];l++) Add(a[l],-1);
            j=P[j];
        }
    }
    printf("%d
",tot);
    for (int i=1;i<=tot;i++) printf("%d
",Ans[i]);
    return 0;
    
}