【Foreign】字符串匹配 [KMP]

字符串匹配

Time Limit: 10 Sec  Memory Limit: 256 MB

Description

　　

Input

　　

Output

　　

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

Sample Output

　　3
　　1 2 4
　　4
　　1 2 3 4
　　3
　　2 3 4

HINT

　　

Solution

　　发现题目中颜色的具体权值是对答案无关的，然后就是只要相对位置一样即可。

　　那么显然是一个KMP的模型，变相更改，条件改为：两个字符上一次出现的相对位置相同（也就是位置差值相等）。

　　那么我们只要求出差值来KMP即可，再考虑一下串长对于答案的影响，差值>串长显然是不会影响答案的，但是直接匹配的话可能将这种情况默认为不可行，所以我们在匹配的时候判断一下串长，若dist>=当前匹配串长则同设为0即可，更新fail的时候也要这么做。这样做KMP即可求出答案。

Code

 #include<iostream>
 #include<string>
 #include<algorithm>
 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<ctime>
 #include<cmath>
 using namespace std;  

 const int ONE=;

 int T,C;
 int n,m;
 int a[ONE],b[ONE];
 int da[ONE],db[ONE],la[ONE],lb[ONE];
 int fail[ONE],j;
 int Ans[ONE],ans_num;

 int get()
 {
         int res=,Q=;char c;
         while( (c=getchar())< || c> )
         if(c=='-')Q=-;
         res=c-;
         while( (c=getchar())>= && c<= )
         res=res*+c-;
         return res*Q;
 }

 int Da(int i,int len) {if(da[i]>=len) return ;return da[i];}
 int Db(int i,int len) {if(db[i]>=len) return ;return db[i];}

 int main()
 {
         T=get();    C=get();
         while(T--)
         {
             n=get();    m=get();
             memset(la,,sizeof(la));    memset(lb,,sizeof(lb));
             for(int i=;i<=n;i++) a[i]=get();
             for(int i=;i<=m;i++) b[i]=get();

             for(int i=;i<=n;i++) da[i]=i-la[a[i]], la[a[i]]=i;
             for(int i=;i<=m;i++) db[i]=i-lb[b[i]], lb[b[i]]=i;

             j=;
             for(int i=;i<=m;i++)
             {
                 j=fail[i-];
                 while(j && Db(j+,j+)!=Db(i,j+)) j=fail[j];
                 if(Db(j+,j+) == Db(i,j+)) j++; else j=;
                 fail[i] = j;
             }

             j=;    ans_num=;
             for(int i=;i<=n;i++)
             {
                 while(j && Db(j+,j+)!=Da(i,j+)) j=fail[j];
                 if(Db(j+,j+) == Da(i,j+)) j++; else j=;
                 if(j==m) Ans[++ans_num] = i-m+, j=fail[j];

             }

             printf("%d\n",ans_num);
             for(int i=;i<=ans_num;i++)
                 printf("%d ",Ans[i]);
             printf("\n");
         }
 }