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; }
时间: 2024-10-13 18:03:08