poj3167- Cow Patterns

传送门

两个串相等定义为串中每一位排序后的相对大小相等。

一位相等等价于这一位前面比他小的和等于他的数的个数相等。

那么用kmp，比较的时候比较这两个个数就可以了。

一开始很瓜地想，询问一段区间内比我小和和我相等的数，得写个主席树啊。。。

实际上用个树状数组维护，kmp跑nxt的时候把跳过的部分从树状数组中删除就好了。

//Achen
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<vector>
#include<queue>
#include<cmath>
#define For(i,a,b) for(int i=(a);i<=(b);i++)
#define Rep(i,a,b) for(int i=(a);i>=(b);i--)
const int N=;
typedef long long LL;
using namespace std;
int n,k,S,s[N],a[N],nxt[N],l1[N],l2[N],ans[N];

template<typename T> void read(T &x) {
    T f=; x=; char ch=getchar();
    while(ch!='-'&&(ch<''||ch>'')) ch=getchar();
    if(ch=='-') f=-,ch=getchar();
    for(;ch>=''&&ch<='';ch=getchar()) x=x*+ch-''; x*=f;
}

int sum[];
void add(int x,int v) {
    if(!x) return;
    for(int i=x;i<=;i+=(i&(-i)))
        sum[i]+=v;
}

int qry(int x) {
    int rs=;
    for(int i=x;i;i-=(i&(-i)))
        rs+=sum[i];
    return rs;
}

void make_nxt(int n) {
    memset(sum,,sizeof(sum));
    For(i,,n-) {
        l1[i]=qry(a[i]-);
        l2[i]=qry(a[i]);
        add(a[i],);
    }
    memset(sum,,sizeof(sum));
    for(int i=,k=;i<n;i++) {
        while(k&&((qry(a[i]-)!=l1[k])||(qry(a[i])!=l2[k]))) {
            For(j,i-k,i-nxt[k-]-) add(a[j],-);
            k=nxt[k-];
        }
        if((qry(a[i]-)==l1[k])||(qry(a[i])==l2[k])) k++;
        nxt[i]=k;
        add(a[i],);
    }
}

void solve(int n,int m) {
    int k=; ans[]=;
    memset(sum,,sizeof(sum));
    For(i,,n-) {
        while(k&&((qry(s[i]-)!=l1[k])||(qry(s[i])!=l2[k]))) {
            For(j,i-k,i-nxt[k-]-) add(s[j],-);
            k=nxt[k-];
        }
        if((qry(s[i]-)==l1[k])||(qry(s[i])==l2[k])) k++;
        if(k==m) {
            ans[++ans[]]=i-m+;
            For(j,i-k+,i-nxt[k-]) add(s[j],-);
            k=nxt[k-];
        }
        add(s[i],);
    }
    For(i,,ans[]) printf("%d\n",i==?ans[i]:ans[i]+);
}

int main() {
#ifdef DEBUG
    freopen("1.in","r",stdin);
    freopen("1.out","w",stdout);
#endif
    while(scanf("%d %d %d",&n,&k,&S)==) {
        For(i,,n-) read(s[i]);
        For(i,,k-) read(a[i]);
        make_nxt(k);
        solve(n,k);
    }
    return ;
}
/*
10 3 10
7
8
4
9
6
4
5
10
4
8

10
9
3
*/