jzoj5603 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

差一点点就在考场上想出来了
对于一个数，将他的值更改为离他最近的前面一个点与他的距离
如果没有就是-1
那么，两个串匹配的条件就是两个串一样
当然，-1要处理一下
如果j+1是-1，那么j+1能和i匹配的条件就是j+1<=a[i]
再套KMP就好了

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int N=1000005;
int b[N],d[N],tot,cnt;
int ans,w[N],a[N],c[N],n,m,i,j,t,k,fl,cc,T,p[N];
int read(){
    int sum=0;
    char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9') {
        sum=sum*10+c-'0';
        c=getchar();}
    return sum;
}
void kmp(){
    int i,j;
    j=0;
    fo(i,2,m) {
      while (j>0&&c[i]!=c[j+1]&&(!(c[j+1]==-1&&j+1<=c[i]))) j=p[j];
      if (c[j+1]==c[i]||(c[j+1]==-1&&j+1<=c[i])) j++;
      p[i]=j;}
    j=0; 
    fo(i,1,n) {
        while (a[i]!=c[j+1]&&j>0&&(!(c[j+1]==-1&&j+1<=a[i]))) j=p[j];
        if (((c[j+1]==a[i])||(c[j+1]==-1&&j+1<=a[i]))&&j<m) j++;
        if (j==m){
            w[++ans]=i-m+1;
            j=p[j];
        }
    }
}
void write(int x)
{
    if (x>9) write(x/10);
    putchar(x%10+'0');
}
int main(){
    freopen("xjz.in","r",stdin);
    freopen("xjz.out","w",stdout);
    T=read(),cc=read();
    while (T){
        T--;
        ans=0;
        memset(b,0,sizeof(b));
        memset(d,0,sizeof(d));
        memset(p,0,sizeof(p));
        n=read(),m=read();
        fo(i,1,n) a[i]=read();
        fo(i,1,m) c[i]=read();
        fo(i,1,n)  
          if (!b[a[i]]) b[a[i]]=i,a[i]=-1;
            else  {
            t=i-b[a[i]],b[a[i]]=i,a[i]=t;
            if (a[i]>m) a[i]=-1;}
        fo(i,1,m) 
          if (!d[c[i]]) d[c[i]]=i,c[i]=-1;
            else  
              t=i-d[c[i]],d[c[i]]=i,c[i]=t;
        kmp();
        write(ans);putchar('\n');
        fo(i,1,ans) write(w[i]),putchar(' ');
        putchar('\n');
    }
}