【JZOJ5603】【NOI2018模拟3.27】Xjz

题目描述

给定字符串 S 和 T。
串A和串B匹配的定义改为：存在一个字符的映射，使得A应用这个映射之后等于B，且这个映射必须为一个排列。
A=121, B=313，当映射为{1->3, 2->1, 3->2}时A’=B,可以匹配
A=212, B=313，当映射为{1->1, 2->3, 3->2}时A’=B,可以匹配
A=232, B=313，当映射为{1->2, 2->3, 3->1}时A’=B,可以匹配
A=123, B=111，当映射为{1->1, 2->1, 3->1}时A’=B,但此时映射不为一个排列，不能匹配
求 S 的哪些连续子串与 T 匹配.

解题思路

两个字符串匹配就是指同构，
考虑如何判断同构，对于一段长度为m的子串，每个位置的值就是，与前面同色的位置的距离。
hash判断。

#include <cmath>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
#include <map>
#include <bitset>
#include <set>
const int inf=2147483647;
const long long mo=1e15+7;  
const int N=1000005;
using namespace std;
int a[N],b[N],n,m,c,T,ans[N],nxt[N],lst[N],co[N];
long long val,ha,mi[N];
void read(int &n)
{
    char ch=' ';int q=0,w=1;
    for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
    if(ch=='-')w=-1,ch=getchar();
    for(;ch>='0' && ch<='9';ch=getchar())q=(q<<1)+(q<<3)+ch-'0';n=q*w;
}
int main()
{
    scanf("%d%d",&T,&c);
    mi[0]=1;
    for(int i=1;i<N;i++) mi[i]=mi[i-1]*101%mo;
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) read(a[i]);
        for(int i=1;i<=m;i++) read(b[i]);
        memset(lst,0,sizeof(lst)),memset(co,0,sizeof(co));
        ha=val=0;
        for(int i=1;i<=m;i++)
        {
            lst[i]=co[b[i]],co[b[i]]=i;
            if(lst[i]) ha=(ha+1ll*(i-lst[i])*mi[m-i]%mo)%mo;
        }
        ans[0]=0;
        memset(nxt,64,sizeof(nxt)),memset(lst,0,sizeof(lst)),memset(co,0,sizeof(co));
        for(int i=1;i<=n;i++)
        {
            lst[i]=co[a[i]],co[a[i]]=i,nxt[lst[i]]=i;
            if(i>m)
                if(nxt[i-m]<i) val=(val-1ll*mi[i-nxt[i-m]-1]*(nxt[i-m]-(i-m))%mo+mo)%mo;
            val=val*mi[1]%mo;
            if(i-m+1<=lst[i])
                if(lst[i]) val=(val+1ll*(i-lst[i]))%mo;
            if(i>=m && val==ha) ans[++ans[0]]=i-m+1;
        }
        printf("%d\n",ans[0]);
        for(int i=1;i<=ans[0];i++) printf("%d ",ans[i]);
        printf("\n");
    }
}