【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=10001;

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=1,Q=1;char c;
        while( (c=getchar())<48 || c>57 )
        if(c=='-')Q=-1;
        res=c-48;
        while( (c=getchar())>=48 && c<=57 )
        res=res*10+c-48;
        return res*Q;
}

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

int main()
{
        T=get();    C=get();
        while(T--)
        {
            n=get();    m=get();
            memset(la,0,sizeof(la));    memset(lb,0,sizeof(lb));
            for(int i=1;i<=n;i++) a[i]=get();
            for(int i=1;i<=m;i++) b[i]=get();
            
            for(int i=1;i<=n;i++) da[i]=i-la[a[i]], la[a[i]]=i;
            for(int i=1;i<=m;i++) db[i]=i-lb[b[i]], lb[b[i]]=i;
            
            j=0;
            for(int i=2;i<=m;i++)
            {
                j=fail[i-1];
                while(j && Db(j+1,j+1)!=Db(i,j+1)) j=fail[j];
                if(Db(j+1,j+1) == Db(i,j+1)) j++; else j=0;
                fail[i] = j;
            }
            
            j=0;    ans_num=0;
            for(int i=1;i<=n;i++)
            {
                while(j && Db(j+1,j+1)!=Da(i,j+1)) j=fail[j];
                if(Db(j+1,j+1) == Da(i,j+1)) j++; else j=0;
                if(j==m) Ans[++ans_num] = i-m+1, j=fail[j];
                
            }
            
            printf("%d\n",ans_num);
            for(int i=1;i<=ans_num;i++)
                printf("%d ",Ans[i]);
            printf("\n");
        }
}