特殊位置kmp匹配--bzoj4641: 基因改造

传送门！
将每个数变成当前位置减上一次出现的位置，用$kmp$匹配，但要注意如果当前位置减上一次出现位置的差超过了匹配长度则视为没有出现过，要特判

#include
#include
#include
#include
#include
#define maxn 1000005
#define LL long long
using namespace std;
int t,n,m,c,a[maxn],b[maxn],pos[maxn],ans[maxn],nxt[maxn];
const int bas=1000003,mod=1e9+7;

inline int rd(){
	int x=0,f=1;char c=' ';
	while(c<'0' || c>'9') f=c=='-'?-1:1,c=getchar();
	while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
	return x*f; 
}

inline bool judge(int x,int y){
	return b[x]==y||(b[x]==-1 && (x<=y||y==-1));//特判当前数字的位置差>=匹配长度 
}

int main(){
	t=rd(); c=rd();
	while(t--){
		n=rd(); m=rd(); int tot=0;
		memset(pos,0,sizeof pos);
		for(int i=1;i<=n;i++){
			int x=rd();
			if(!pos[x]) a[i]=-1,pos[x]=i;
			else a[i]=i-pos[x],pos[x]=i;
		}
		memset(pos,0,sizeof pos);
		for(int i=1;i<=m;i++){
			int x=rd();
			if(!pos[x]) b[i]=-1,pos[x]=i;
			else b[i]=i-pos[x],pos[x]=i;
		}
		for(int i=2,j=0;i<=m;i++){
			while(j>0 && !judge(j+1,b[i])) j=nxt[j];
			if(judge(j+1,b[i])) ++j;
			nxt[i]=j; 
		}
		for(int i=1,j=0;i<=n;i++){
			while(j>0 && !judge(j+1,a[i])) j=nxt[j];
			if(judge(j+1,a[i])) ++j;
			if(j==m) ans[++tot]=i-m+1,j=nxt[j];
		}
		printf("%d\n",tot);
		for(int i=1;i<=tot;i++) printf("%d ",ans[i]); puts("");
	}
	return 0;
}