bzoj 4503 两个串

Description

兔子们在玩两个串的游戏。给定两个字符串S和T，兔子们想知道T在S中出现了几次，

分别在哪些位置出现。注意T中可能有“?”字符，这个字符可以匹配任何字符。

Input

两行两个字符串，分别代表S和T

Output

第一行一个正整数k，表示T在S中出现了几次

接下来k行正整数，分别代表T每次在S中出现的开始位置。按照从小到大的顺序输出，S下标从0开始。

Sample Input

ababcadaca
a?a

Sample Output

3
0
5

HINT

S 长度不超过 10^5， T 长度不会超过 S。 S 中只包含小写字母， T中只包含小写字母和“?”

将b串反转

设两个串的距离值为∑(a_i-a_j)²a_ia_j=∑a_i³a_j+∑a_ia_j³-∑2a_i²a_j²

当为*时a_i为0

FFT直接上

#include<bits/stdc++.h>
#define pi acos(-1)
#define maxn 300010
using namespace std;
struct Complex{
    double r,i;
    Complex(double x=,double y=){r=x,i=y;}
    Complex operator + (Complex x){return Complex(r+x.r,i+x.i);}
    Complex operator - (Complex x){return Complex(r-x.r,i-x.i);}
    Complex operator * (Complex x){return Complex(r*x.r-i*x.i,r*x.i+i*x.r);}
}A[maxn],B[maxn],C[maxn];
char a[maxn],b[maxn];
int rev[maxn];
int N,L;
int k1[maxn],k2[maxn];
int l1,l2;
int ans;
int k[maxn];
void FFT(Complex a[],int n,int f){
    for(int i=;i<n;++i)
        if(i<rev[i])
            swap(a[i],a[rev[i]]);
    for(int i=;i<=n;i<<=){
        Complex wn=Complex(cos(*pi/i),sin(*pi/i)*f);
        for(int j=;j<n;j+=i){
            Complex w=Complex(,);
            for(int k=;k<(i>>);++k,w=w*wn){
                Complex tmp1=a[j+k],tmp2=a[j+k+(i>>)]*w;
                a[j+k]=tmp1+tmp2;a[j+k+(i>>)]=tmp1-tmp2;
            }
        }
    }
    if(f==-)for(int i=;i<n;++i)a[i].r/=n;
}
void init(){
    scanf("%s%s",a,b);
    l1=strlen(a);l2=strlen(b);
    for(int i=;i<l1;++i)k1[i]=a[i]-'a'+;
    for(int i=;i<(l2>>);++i)swap(b[i],b[l2-i-]);
    for(int i=;i<l2;++i)
        if(b[i]=='?')k2[i]=;
        else k2[i]=b[i]-'a'+;
    N=,L=;while(N<l1+l2)N<<=,L++;
    for(int i=;i<N;++i)rev[i]=(rev[i>>]>>)|((i&)<<(L-));
}
void work(){
    for(int i=;i<N;++i)A[i]=Complex(k1[i],);
    for(int i=;i<N;++i)B[i]=Complex(k2[i]*k2[i]*k2[i],);
    FFT(A,N,);FFT(B,N,);
    for(int i=;i<N;++i)C[i]=C[i]+(A[i]*B[i]);
    for(int i=;i<N;++i)A[i]=Complex(k1[i]*k1[i]*k1[i],);
    for(int i=;i<N;++i)B[i]=Complex(k2[i],);
    FFT(A,N,);FFT(B,N,);
    for(int i=;i<N;++i)C[i]=C[i]+(A[i]*B[i]);
    for(int i=;i<N;++i)A[i]=Complex(k1[i]*k1[i]*,);
    for(int i=;i<N;++i)B[i]=Complex(k2[i]*k2[i],);
    FFT(A,N,);FFT(B,N,);
    for(int i=;i<N;++i)C[i]=(C[i]-(A[i]*B[i]));
    FFT(C,N,-);
    for(int i=l2-;i<l1;++i)
        if(C[i].r<0.5)k[++ans]=i-l2+;
    printf("%d\n",ans);
    for(int i=;i<=ans;++i)
        printf("%d\n",k[i]);
}
int main(){
    init();
    work();
    return ;
}