【BZOJ4641】基因改造
Description
"人类智慧的冰峰，只有萌萌哒的我寂寞地守望。"
--TB
TB正走在改造人类智慧基因的路上。TB发现人类智慧基因一点也不萌萌哒，导致人类普遍智商过低，为了拯救低智商人群，博爱的TB开始改造人类智慧基因。人类智慧DNA由C种人类智慧脱氧核苷酸构成（这是一种十分特殊的DNA）。TB智慧DNA片段T长度为M，她可以把另一段长度为M的人类智慧DNA片段S改造成T。改造过程中，TB可以充分发挥智慧，将任意两种人类智慧脱氧核苷酸交换（比如对于片段S=12321，交换1和2变成S=21312，交换1和4变成42324），可以无限次交换。如果S可以通过若干次交换变成T，那么就称S为"萌萌哒人类基因片段"。TB想知道对于一个长度为N的人类智慧DNA片段S[1~N]，有多少个长度为M的连续子片段S[i~i+M-1]，是"萌萌哒人类基因片段"，并且这些"萌萌哒人类基因片段"在哪里。
Input
输入文件的第一行包含两个正整数case和C，分别表示数据组数和人类智慧脱氧核苷酸的种数。
接下来3*case行，每三行表示一组数据：
第一行一个正整数N和M，表示人类智慧DNA片段S和TB智慧DNA片段T的长度。
第二行N个正整数，表示人类智慧DNA片段S。
第三行M个正整数，表示TB智慧DNA片段T。
对于所有数据数据，case=3, n,m,C<=1000000
Output
对于每组数据：
第一行一个正整数tot，表示"萌萌哒人类基因片段"的个数。
接下来一行tot个用空格隔开的正整数pos，表示"萌萌哒人类基因片段"开头所在的位置。要求从小到大输出每个pos。
Sample Input
3 3
6 3
1 2 1 2 3 2
3 1 3
6 3
1 2 1 2 1 2
3 1 3
6 3
1 1 2 1 2 1
3 1 3
6 3
1 2 1 2 3 2
3 1 3
6 3
1 2 1 2 1 2
3 1 3
6 3
1 1 2 1 2 1
3 1 3
Sample Output
3
1 2 4
4
1 2 3 4
3
2 3 4
对于第一组数据：
S[1~3]=121，可以先将1和2交换变成212，再将2和3交换变成313。
S[2~4]=212，可以将2和3交换变成313。
S[4~6]=232，可以先将2和3交换变成323，再将1和2交换变成313。
1 2 4
4
1 2 3 4
3
2 3 4
对于第一组数据：
S[1~3]=121，可以先将1和2交换变成212，再将2和3交换变成313。
S[2~4]=212，可以将2和3交换变成313。
S[4~6]=232，可以先将2和3交换变成323，再将1和2交换变成313。
题解：这种题多做做就差不多能掌握套路了。
显然我们将KMP的比较函数改一改即可，不难想到用pre[i]，即i上一次出现的位置来当做判相等的条件。特别地，如果pre[i]超出了S与T匹配的范围，则我们视为pre[i]=-1。细节还是需要注意一下的~
#include <cstdio>
#include <cstring>
#include <iostream> using namespace std;
const int maxn=1000010;
int cas,n,m,C,tot;
int next[maxn],p1[maxn],p2[maxn],pos[maxn],last[maxn];
inline int rd()
{
int ret=0,f=1; char gc=getchar();
while(gc<'0'||gc>'9') {if(gc=='-') f=-f; gc=getchar();}
while(gc>='0'&&gc<='9') ret=ret*10+gc-'0',gc=getchar();
return ret*f;
}
void work()
{
n=rd(),m=rd(),tot=0;
int i,j,a;
memset(last,-1,sizeof(last));
for(i=0;i<n;i++) a=rd(),p1[i]=(last[a]==-1)?-1:i-last[a],last[a]=i;
memset(last,-1,sizeof(last));
for(i=0;i<m;i++) a=rd(),p2[i]=(last[a]==-1)?-1:i-last[a],last[a]=i;
next[0]=-1,i=0,j=-1;
p1[n]=p2[m]=0;
while(i<m)
{
if(j==-1||p2[i]==p2[j]||(p2[i]>j&&p2[j]==-1)) next[++i]=++j;
else j=next[j];
}
i=j=0;
while(i<n)
{
if(j==-1||p1[i]==p2[j]||(p1[i]>j&&p2[j]==-1)) i++,j++;
else j=next[j];
if(j==m) pos[++tot]=i-m+1;
}
printf("%d\n",tot);
for(i=1;i<=tot;i++) printf("%d ",pos[i]);
printf("\n");
}
int main()
{
cas=rd(),C=rd();
while(cas--) work();
return 0;
}//3 3 6 3 1 2 1 2 3 2 3 1 3 6 3 1 2 1 2 1 2 3 1 3 6 3 1 1 2 1 2 1 3 1 3