【poj 3167】Cow Patterns（字符串--KMP匹配+数据结构--树状数组）

题意：给2个数字序列 a 和 b ，问按从小到达排序后，a中的哪些子串与b的名次匹配。 a 的长度 N≤100,000，b的长度 M≤25,000，数字的大小 K≤25。

解法：【思考】1.X 暴力。枚举 a 中的子串，选出来排序后比对名次。O(n*  m log m  *m)=O(n*m^2*log m)。
    2.X  暴力+树状数组优化。枚举 a 中的子串，选出来后比较前缀名次（P.S.这种转化常出现，比如：【洛谷 p3368】模板-树状数组 2（数据结构） 就是将个体转化为前缀。），看加上自己的之前小于等于和等于该数的个数是否相等。O(n*m log m)。
    3.√  kmp+树状数组优化。直接kmp匹配 a串和 b串，next[ ]预处理 b串，再在kmp时看前缀名次来比较。（这个要kmp理解得很好才能在处理树状数组时处理得很好，我现在思绪混乱啊......代码就算了......qwq）

 

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;

const int N=100010,M=25010,K=30;
struct node{int x,y,d;}b[M],a[N];
int a[N],b[M],next[M];
int cnt=0,s[N];
int ca[K],cb[K];

int lowbit(int x) {return x&-x;}
int change(int x,int d)
{
    for (int i=x;i<=n;i+=lowbit(i))
      ca[i]+=d;
}
int getcount(int x)
{
    int sum=0;
    for (int i=x;i>=1;i-=lowbit(i))
      sum+=ca[i];
    return sum;
}
int change_2(int x,int d)
{
    for (int i=x;i<=m;i+=lowbit(i))
      cb[i]+=d;
}
int getcount_2(int x)
{
    int sum=0;
    for (int i=x;i>=1;i-=lowbit(i))
      sum+=cb[i];
    return sum;
}
void init_kmp()
{
    memset(cb,0,sizeof(cb));
    memset(next,0,sizeof(next));
    int p=0,x=0,y=0;
    next[1]=0;
    for (int i=2;i<=m;i++)
    {
      while ((b[i].x!=b[p+1].x||b[i].y!=b[p+1].y) && p)
      {
        p=next[p];
      }
      if (b[i].x==b[p+1].x&&b[i].y==b[p+1].y)
      {
        p++;
        add_2(b[i].d,1);
      }
      next[i]=p;
    }
}
void kmp()
{
    int p=0;
    for (int i=1;i<=n;i++)
    {
      int x=getcount_2(a[i].d-1);
      while ((a[i].x-(a[i-p].x+a[i-p].y)!=b[p+1].x||a[i].y!=b[p+1].y) && p) p=next[p];
      if (a[i].x==b[p+1].x&&a[i].y==b[p+1].y) p++;
      if (p==m) s[++cnt]=i;
    }
}
int main()
{
    freopen("a.in","r",stdin);
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    //memset(ca,0,sizeof(ca));
    for (int i=1;i<=n;i++)
    {
      scanf("%d",&a[i].d);
      /*change(a[i].d,1);
      a[i].x=getcount(a[i].d);
      a[i].y=a[i].x-getcount(a[i].d-1);*/
    }
    //memset(cb,0,sizeof(cb));
    for (int i=1;i<=m;i++)
    {
      scanf("%d",&b[i].d);
      /*change_2(b[i].d,1);
      b[i].x=getcount_2(b[i].d);
      b[i].y=b[i].x-getcount_2(b[i].d-1);*/
    }
    init_kmp();
    kmp();
    printf("%d\n",cnt);
    for (int i=1;i<=cnt;i++)
      printf("%d ",s[i]);
    return 0;
}