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');
    }
}