- 题意
给定一个顺序序列表示奶牛的排列顺序,定义串a=b当且仅当a的每位的排名=b的每位的排名。 - 题解
预处理出前i位中j出现的次数,由于编号很小,处理出类似前缀和,暴力查询排名即可。已知排名之后就直接kmp。 - 代码
#include <iostream>
#include <cstdio>
#include <cstring>
#define MAXN 100001
using namespace std;
int n,m,s;
int s1[MAXN],s2[MAXN];
int p1[MAXN][30],p2[MAXN][30];
void init(){
for(int i=1;i<=n;i++){
for(int j=1;j<=s;j++)
p1[i][j]=p1[i-1][j];
p1[i][s1[i]]++;
}
for(int i=1;i<=m;i++){
for(int j=1;j<=s;j++)
p2[i][j]=p2[i-1][j];
p2[i][s2[i]]++;
}
}
int nxt[MAXN];
void getnxt(){
int fix=0;
int i=1;
while(i<=m){
int u1=0,u2=0;
int a1=0,a2=0;
for(int j=1;j<s2[i];j++)//i fix+1
u1+=p2[i][j]-p2[i-fix][j];
a1=p2[i][s2[i]]-p2[i-fix][s2[i]];
for(int j=1;j<s2[fix];j++)
u2+=p2[fix][j];
a2=p2[fix][s2[fix]];
if(fix==0||(a1==a2&&u1==u2)) i++,fix++,nxt[i]=fix;
else fix=nxt[fix];
}
}
int ans[MAXN],tot;
void kmp(){
int i=1,j=1;
while(i<=n){
int u1=0,u2=0;
int a1=0,a2=0;
for(int k=1;k<s1[i];k++)
u1+=p1[i][k]-p1[i-j][k];
a1=p1[i][s1[i]]-p1[i-j][s1[i]];
for(int k=1;k<s2[j];k++)
u2+=p2[j][k];
a2=p2[j][s2[j]];
if(!j||(a1==a2&&u1==u2)) j++,i++;
else j=nxt[j];
if(j==m+1){
ans[++tot]=i-m;
j=nxt[j];
}
}
}
int main(){
while(~scanf("%d%d%d",&n,&m,&s)){
memset(nxt,0,sizeof nxt);
memset(p1,0,sizeof p1);
memset(p2,0,sizeof p2);
tot=0;
for(int i=1;i<=n;i++) scanf("%d",&s1[i]);
for(int i=1;i<=m;i++) scanf("%d",&s2[i]);
init();
getnxt();
kmp();
printf("%d\n",tot);
for(int i=1;i<=tot;i++)
printf("%d\n",ans[i]);
}
return 0;
}