牛客NC15879 A Simple Problem

传送门：A Simple Problem

题意

给定两个序列s1和s2，同样的数字可以用相同的别的数字代替（并且也可以是出现过的数字），问s2在s1中出现了几次。

题解

首先预处理一下这两个序列，因为数字的位置是不会变得，所以把每个数字用当前位置和前一次出现的位置的差表示，如果是第一次出现，用-1表示，这里为什么是-1呢，不能比-1更大，但可以更小，不然如果匹配的时候 j=0，那么如果 s[i]=0 这样判断的就是不是该满足的那个条件了，这样是不对的。然后就是kmp匹配了，匹配的判断条件改一下就好了，如果当前匹配上了的长度 j 比当前数字和前一次出现的位置的差( s[i] )小的话，并且当前准备匹配的那个数字( s[j] )是第一次出现，那么是可以匹配上的，相当于是把 s[j] 的数字表示成了s[i] 的数字，这样差就是一样的了。如果 j>=s[i] 的话，说明如果s[i]==s[j] 表示他们和前一次出现的位置距离是相等的，这样也是可以匹配上的。如果匹配不上，那么就 j=nt[i]；

代码

#include<bits/stdc++.h>
#define ull unsigned long long
#define ll long long
#define pb push_back
#define ft first
#define sd second
#define pii pair<int,int>
#define pll pair<ll,ll>
using namespace std;
 
const int maxn=2e5+10;
int s1[maxn],s2[maxn],nt[maxn],pre[maxn],pos2[maxn],pos1[maxn];
 
void getpre(int s[],int pos[],int n,int k)
{
    for(int i=0;i<k;i++) pre[i]=-1;
    for(int i=0;i<n;i++){
        if(pre[s[i]]==-1) pos[i]=-1;
        else pos[i]=i-pre[s[i]];
        pre[s[i]]=i;
    }
}
 
void getnt(int s[],int n)
{
    int i=0,j=-1;
    nt[0]=-1;
    while(i<n){
        if(j==-1||(j<s[i]?s[j]==-1:s[i]==s[j])) i++,j++,nt[i]=j;
        else j=nt[j];
    }
}
 
int kmp(int s[],int p[],int n,int m)
{
    getnt(p,m);
    int i=0,j=0,ans=0;
    while(i<n){
        if(j==-1||(j<s[i]?p[j]==-1:s[i]==p[j])) i++,j++;
        else j=nt[j];
        if(j==m) ans++;
    }
    return ans;
}
 
int main()
{
    int n,k,m;
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++) scanf("%d",&s1[i]);
    scanf("%d",&m);
    for(int i=0;i<m;i++) scanf("%d",&s2[i]);
    getpre(s1,pos1,n,k);
    getpre(s2,pos2,m,k);
    printf("%d
",kmp(pos1,pos2,n,m));
    return 0;
}