JZOJ5603. 【NOI2018模拟3.27】Xjz

Description

给定字符串 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 匹配.

Input

第一行两个整数 T(T<=3),C, 分别表示数据组数与字符集大小.
对于每组数据, 第一行两个整数 n,m, 分别表示 S,T 的长度.
第二行 n 个整数, 第 i 个整数表示 S i .
第三行 m 个整数, 第 i 个整数表示 T i .

Output

对于每组数据输出两行, 第一行一个整数 k 表示匹配的个数.
第二行 k 个整数表示匹配的子串在 S 中的开始位置 (下标从 1 开始), 升序排列.

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

Data Constraint

对于前 10% 的数据, n,m,C ≤ 1000;
对于前 30% 的数据, n,m ≤ 100000, C ≤ 40;
对于前 60% 的数据, n,m,C ≤ 100000;
对于 100% 的数据, n,m,C ≤ 1000000.

题解

S与T匹配就等同于它们每个位置距离相邻的与它们相同的数的距离相同。
而T是固定的，用哈希。

匹配的时候要注意一个问题，
就是S中，在某个前面的最近的可能超出了枚举到的开头位置，
那么就应该将这个位置的距离设为0。

code

#include <queue>
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#include <time.h>
#define ll long long
#define N 1000003
#define M 103
#define db double
#define P putchar
#define G getchar
#define mo 998244353
using namespace std;
char ch;
void read(int &n)
{
    n=0;
    ch=G();
    while((ch<'0' || ch>'9') && ch!='-')ch=G();
    ll w=1;
    if(ch=='-')w=-1,ch=G();
    while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();
    n*=w;
}

int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
ll abs(ll x){return x<0?-x:x;}
ll sqr(ll x){return x*x;}
void write(ll x){if(x>9) write(x/10);P(x%10+'0');}

int s[N],t[N],k;
int T,c,n,m,lst[N],p1[N],p2[N];
int ans[N],ss[N],tt[N],p[N],x,y;
ll sum,S,z[N];
bool pd;

void get(int n,int* s,int* nxt)
{
    memset(lst,0,sizeof(lst));
    for(int i=n;i;i--)
        nxt[i]=lst[s[i]],lst[s[i]]=i;
}

int main()
{
    read(T),read(c);
    for(;T;T--)
    {
        read(n);read(m);
        z[0]=1;ans[0]=0;
        for(int i=1;i<=1000000;i++)
            z[i]=z[i-1]*n%mo;
        for(int i=1;i<=n;i++)read(s[i]);
        for(int i=1;i<=m;i++)read(t[i]);
        get(n,s,p1);get(m,t,p2);

        memset(ss,0,sizeof(ss));
        memset(tt,0,sizeof(tt));
        for(int i=1;i<=n;i++)ss[p1[i]]=i;
        for(int i=1;i<=m;i++)tt[p2[i]]=i;

        sum=S=0;

        for(int i=1;i<=m;i++)
            p[i]=tt[i]?i-tt[i]:0,sum=(sum*n%mo+p[i])%mo;

        for(int i=1;i<m;i++)
            p[i]=ss[i]?i-ss[i]:0,S=(S*n%mo+p[i])%mo;

        for(int i=m;i<=n;i++)
        {
            x=i-m;
            p[i]=ss[i]>x?i-ss[i]:0;
            S=(S-p[x]*z[m-1]%mo+mo)%mo;
            if(p1[x]<=i)
            {
                if(p1[x]<i)S=(S-z[i-p1[x]-1]*p[p1[x]]%mo+mo)%mo;
                p[p1[x]]=0;
            }
            S=(S*n%mo+p[i])%mo;
            if(S==sum)ans[++ans[0]]=x+1;
        }

        write(ans[0]);P('\n');
        for(int i=1;i<=ans[0];i++)
            write(ans[i]),P(' ');
        P('\n');
    }

    return 0;
}