JZOJ 5603. 【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.

Solution

• 由于两个串的匹配方式是不确定的，我们需要找到一个通用的方法来匹配两个串。

• 我们考虑处理出每个字符的前一个字符距离自己的位置 $w_i$ 。

• 如果两个串的 $w$ 数组相同，则两个串可以匹配！（通用的方法！）

• 若一个字符之前没有与之相同的字符，则其 $w$ 值为 $-1$ 。

• 注意：若模式串T的某一位为 $-1$ ，而匹配串S的某一位的前一字符已跨过本串，他们仍能匹配。

• 如此，我们直接做 KMP 即可，时间复杂度 $O(T*N)$ 。

Code

#include<cstdio>
#include<cstring>
#include<cctype>
using namespace std;
const int N=1e6+5;
int ws[N],wt[N],next[N],f[N];
inline int read()
{
    int X=0,w=0; char ch=0;
    while(!isdigit(ch)) w|=ch=='-',ch=getchar();
    while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
inline void write(int x)
{
    if(x>9) write(x/10);
    putchar(x%10+'0');
}
int main()
{
    int T=read();read();
    while(T--)
    {
        int n=read(),m=read();
        memset(f,0,sizeof(f));
        for(int i=1;i<=n;i++)
        {
            int ch=read();
            ws[i]=f[ch]?i-f[ch]:-1;
            f[ch]=i;
        }
        memset(f,0,sizeof(f));
        for(int i=1;i<=m;i++)
        {
            int ch=read();
            wt[i]=f[ch]?i-f[ch]:-1;
            f[ch]=i;
        }
        for(int i=2,j=0;i<=m;i++)
        {
            while(j && !(wt[i]==wt[j+1] || wt[j+1]<0 && wt[i]>=j+1)) j=next[j];
            if(wt[i]==wt[j+1] || wt[j+1]<0 && wt[i]>=j+1) j++;
            next[i]=j;
        }
        for(int i=1,j=f[0]=0;i<=n;i++)
        {
            while(j && !(ws[i]==wt[j+1] || wt[j+1]<0 && ws[i]>=j+1)) j=next[j];
            if(ws[i]==wt[j+1] || wt[j+1]<0 && ws[i]>=j+1) j++;
            if(j==m)
            {
                f[++f[0]]=i-m+1;
                j=next[j];
            }
        }
        write(f[0]),putchar('\n');
        for(int i=1;i<=f[0];i++) write(f[i]),putchar(' ');
        putchar('\n');
    }
    return 0;
}