BZOJ 1729: [Usaco2005 dec]Cow Patterns 牛的模式匹配

Description

约翰的N(1≤N≤100000)只奶牛中出现了K(1≤K≤25000)只爱惹麻烦的坏蛋．奶牛们按一定的顺序排队的时候，这些坏蛋总会站在一起．为了找出这些坏蛋，约翰让他的奶牛排好队进入牛棚，同时需要你的慧眼来识别坏蛋,为了区分，约翰给所有奶牛都发了号牌，上面写着一个1．．S(1≤S≤25)之间的数字．虽然这不是一个完美的方法，但也能起一点作用．现在，约翰已经不记得坏蛋们的具体号码．但是凭他的记忆，他给出一个“模式串”．原坏蛋的号码如果相同，模式串中他们的号码依然相同．模式串中坏蛋们之间号码的大小关系也与原号码相同的．比如，对于这样一个模式串：1,4,4,3,,2,1 。原来的6只坏蛋，排最前面的与排最后的号码相同（尽管不一定是1），而且他们的号码在团伙中是最小的．第2，3位置的坏蛋，他们的号码也相同（不一定是4），且是坏蛋团伙中最大的． 现在所有奶牛排成队列，号码依次是这样： 5，6，2，10，10，7，3，2，9存在子串2，10，10，7，3，2，满足模式串的相同关系和大小关系，所以这就是坏蛋团伙, 请找出K个坏蛋的困伙的所有可能性．

Input

第1行输入三个整数N，K，S．接下来N行每行输入一只牛的号码．接下来K行每行输入一个模式串的号码．

Output

第1行输出一个整数B.接下来B行，每行一个整数，表示一种可能下的坏蛋团伙的起始位置． 题解：

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

//by zrt
//problem:
using namespace std;
typedef long long LL;
const int inf(0x3f3f3f3f);
const double eps(1e-9);
int n,k,s;int aa[100005][26],bb[100005][26];int a[100005],b[100005];
bool cmp(int j,int i,int c1,int c2){//b 0...j | a i-j..i
int sum1=0,sum2=0;
for(int k=1;k<c1;k++){
sum1+=j?bb[j-1][k]:0;
}
for(int k=1;k<c2;k++){
sum2+=(i?aa[i-1][k]:0)-((i-j)?aa[i-j-1][k]:0);
}
return sum1==sum2&&(j?bb[j-1][c1]:0)==((i?aa[i-1][c2]:0)-((i-j)?aa[i-j-1][c2]:0));
}
bool cmpp(int j,int i,int c1,int c2){//b 0...j |b i-j..i
int sum1=0,sum2=0;
for(int k=1;k<c1;k++){
sum1+=j?bb[j-1][k]:0;
}
for(int k=1;k<c2;k++){
sum2+=(i?bb[i-1][k]:0)-((i-j)?bb[i-j-1][k]:0);
}
return sum1==sum2&&(j?bb[j-1][c1]:0)==((i?bb[i-1][c2]:0)-((i-j)?bb[i-j-1][c2]:0));
}
int nxt[100005];

void mk_pre(){
for(int i=0;i<n;i++) {
if(i)for(int j=1;j<=s;j++) aa[i][j]=aa[i-1][j];
aa[i][a[i]]++;
}
for(int i=0;i<k;i++) {
if(i)for(int j=1;j<=s;j++) bb[i][j]=bb[i-1][j];
bb[i][b[i]]++;
}
}
void mk(){
int j=-1;
nxt[0]=-1;
for(int i=1;i<k;i++){
while(~j&&!cmpp(j+1,i,b[j+1],b[i])) j=nxt[j];
if(cmpp(j+1,i,b[j+1],b[i])) j++;
nxt[i]=j;
}
}

int ans[100005],top;
int main(){
#ifdef LOCAL
freopen("in.txt","r",stdin);
freopen("out.txt","w",stdout);
#endif
scanf("%d%d%d",&n,&k,&s);
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
}
for(int i=0;i<k;i++){
scanf("%d",&b[i]);
}
mk_pre();// qianzhuihe
mk();
int j=-1;
for(int i=0;i<n;i++){
while(~j&&!cmp(j+1,i,b[j+1],a[i])) j=nxt[j];
if(cmp(j+1,i,b[j+1],a[i])) j++;
if(j==k-1){
ans[top++]=i-j+1;
j=nxt[j];
}
}
printf("%d\n",top);
for(int i=0;i<top;i++){
printf("%d\n",ans[i]);
}
return 0;
}