bzoj4503 两个串

令$'?'=0$，那么两个位置可以匹配就可以看成$(s_i-t_j)^2t_j=0$拆开以后把反序，得到三个卷积的和，FFT就可以了。

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define LL long long
const int p=998244353,g=3,maxn=1000010;
int inc(int x,int y)
{
    x+=y;
    return x>=p?x-p:x;
}
int dec(int x,int y)
{
    x-=y;
    return x<0?x+p:x;
}
int pow(int b,int k)
{
    int r=1;
    for (;k;k>>=1,b=(LL)b*b%p)
        if (k&1) r=(LL)r*b%p;
    return r;
}
char s[maxn],t[maxn];
int a[maxn],b[maxn],ans[maxn],rev[maxn],pw[maxn],n,m,l,u,inv;
void fft(int *a,int fl)
{
    int x,t1,t2;
    for (int i=0;i<l;i++)
        if (rev[i]>i) swap(a[rev[i]],a[i]);
    for (int i=0;i<u;i++)
        for (int j=0;j<l;j+=1<<(i+1))
        {
            x=0;
            for (int k=j;k<j+(1<<i);k++)
            {
                t1=a[k];
                t2=(LL)a[k+(1<<i)]*pw[x]%p;
                a[k]=inc(t1,t2);
                a[k+(1<<i)]=dec(t1,t2);
                x+=fl*(1<<(u-i-1));
                if (x<0) x+=l;
            }
        }
    if (fl==-1)
        for (int i=0;i<l;i++)
            a[i]=(LL)a[i]*inv%p;
}
int main()
{
    //freopen("b.in","r",stdin);
    int res=0;
    scanf("%s%s",s,t);
    n=strlen(s);
    m=strlen(t);
    for (int i=0;i<n;i++) s[i]=s[i]-'a'+1;
    for (int i=0;i<m;i++)
        if (t[i]=='?') t[i]=0;
        else t[i]=t[i]-'a'+1;
    for (int i=0;2*i<m-1;i++) swap(t[i],t[m-i-1]);
    l=1,u=0;
    while (l<m+n-1) l<<=1,u++;
    inv=pow(l,p-2);
    pw[0]=1;
    pw[1]=pow(g,(p-1)/l);
    for (int i=2;i<l;i++) pw[i]=(LL)pw[i-1]*pw[1]%p;
    for (int i=0;i<l;i++)
        for (int j=0;j<u;j++)
            rev[i]|=((i>>j)&1)<<(u-j-1);
    for (int i=0;i<n;i++) a[i]=s[i]*s[i];
    for (int i=0;i<m;i++) b[i]=t[i];
    fft(a,1);
    fft(b,1);
    for (int i=0;i<l;i++) ans[i]=(LL)a[i]*b[i]%p;
    //fft(ans,-1);
    for (int i=0;i<n;i++) a[i]=p-2*s[i];
    for (int i=n;i<l;i++) a[i]=0;
    for (int i=0;i<m;i++) b[i]=t[i]*t[i];
    for (int i=m;i<l;i++) b[i]=0;
    fft(a,1);
    fft(b,1);
    for (int i=0;i<l;i++) ans[i]=inc(ans[i],(LL)a[i]*b[i]%p);
    //for (int i=0;i<l;i++) ans[i]=(LL)a[i]*b[i]%p;
    //fft(ans,-1);
    for (int i=0;i<n;i++) a[i]=1;
    for (int i=n;i<l;i++) a[i]=0;
    for (int i=0;i<m;i++) b[i]=t[i]*t[i]*t[i];
    for (int i=m;i<l;i++) b[i]=0;
    fft(a,1);
    fft(b,1);
    for (int i=0;i<l;i++) ans[i]=inc(ans[i],(LL)a[i]*b[i]%p);
    fft(ans,-1);
    for (int i=m-1;i<n;i++)
        if (!ans[i]) res++;
    printf("%d\n",res);
    for (int i=m-1;i<n;i++)
        if (!ans[i]) printf("%d\n",i-m+1);
}