首页 > 代码库 > BZOJ 1729: [Usaco2005 dec]Cow Patterns 牛的模式匹配

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;}

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