JZOJ 5603 Xjz

Xjz

Description

给定字符串$S$和$T$。
串$A$和串$B$匹配的定义改为：存在一个字符的映射，使得$A$应用这个映射之后等于$B$，且这个映射必须为一个排列。
$A$=$121$,$B=313$，当映射f为{1->3, 2->1, 3->2}时$f(A)=B$,可以匹配
求$S$有多少个子串与$T$匹配。

Data Constraint

$|S|,|T|≤10^6$

Solution

这种题解题方向很明显，多半是$Hash$。
设$w_i$表示与$i$位置上的数相同的前一个位置与它的距离，若两个子串的$w$完全相同，则一定可以匹配，反之一定不可匹配。
那问题就简单了，用个指针表示$T$的第一个位置在$S$的哪个位置上，让指针从左往右扫，同时维护一下$S$的子串的$w$的$Hash$，有些位置的$w$可能会在移动过程中变为$-1$（即前面没有位置匹配），多维护一个$last$数组，按照$last_i$排序，依次修改 w <script type="math/tex" id="MathJax-Element-645">w</script>即可。

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

#define fo(i,j,l) for(int i=j;i<=l;++i)
#define fd(i,j,l) for(int i=j;i>=l;--i)

using namespace std;
typedef long long ll;
const ll N=11e5,K=31,mo=998244353;

int T,C,n,m;
int s[N],t[N],last[N];
int dq[N],ut[N],ans[N];
ll q[N],xs[N];

bool kmp(int a,int b)
{return last[a]<last[b];}

inline int read()
{
    int o=0; char ch=' ';
    for(;ch<'0'||ch>'9';ch=getchar());
    for(;ch>='0'&&ch<='9';ch=getchar())o=o*10+ch-48;
    return o;
}

int main()
{
    cin>>T>>C;
    fo(tt,1,T){
        cin>>n>>m;
        q[1]=1;
        fo(i,1,n)last[i]=dq[i]=0;
        fo(i,2,n)q[i]=q[i-1]*K%mo;
        fo(i,1,n)s[i]=read();
        fo(i,1,m)t[i]=read();
        fo(i,1,n)last[i]=dq[s[i]],dq[s[i]]=i,xs[i]=last[i]!=0?i-last[i]:0;
        fo(i,1,n)dq[i]=0;
        ll hash=0,h1=0;
        fo(i,1,m)hash=(hash*K+i-(dq[t[i]]==0?i:dq[t[i]]))%mo,dq[t[i]]=i;
        fo(i,1,n)ut[i]=i;
        sort(ut+1,ut+n+1,kmp);
        fo(i,1,m)h1=(h1*K+xs[i])%mo;
        int oo=0,w=1;
        if(h1==hash)ans[++oo]=1;
        fo(i,2,n-m+1){
            for(;last[ut[w]]<i&&w<=n;++w){
                if(ut[w]<m+i-1)h1=(h1+mo-q[i+m-1-ut[w]]*xs[ut[w]]%mo)%mo;
                xs[ut[w]]=0;
            }
            h1=(h1+mo-(xs[i-1])*q[m]%mo)%mo;
            h1=h1*K%mo;
            h1=(h1+xs[i+m-1])%mo;
            if(h1==hash)ans[++oo]=i;
        }
        cout<<oo<<endl;
        fo(i,1,oo)printf("%d ",ans[i]);
        puts("");
    }
}