bzoj 4259

与bzoj 4503是一个题,只是两个串都有通配符

如果你没有做过那道题,请看这里

接下来假设你知道这些前置知识了

那么对于两个通配符,我们将表达式变成这样即可:

$\sum_{i=0}^{l}S_i*T_i*(S_i-T_i)^2==0$

然后打开:

$\sum_{i=0}^{l}T_i*(S_i)^3+S_i*(T_i)^3-2*(S_i)^2*(T_i)^2==0$

这次三项都得用卷积处理

#include <cstdio>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <queue>
#include <stack>
#define ll long long
using namespace std;
const double pi=acos(-1.0);
struct cp
{
    double x,y;
    friend cp operator + (cp a,cp b)
    {
        return (cp){a.x+b.x,a.y+b.y};
    }
    friend cp operator - (cp a,cp b)
    {
        return (cp){a.x-b.x,a.y-b.y};
    }
    friend cp operator * (cp a,cp b)
    {
        return (cp){a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x};
    }
};
char ssa[1000005],ssb[1000005];
int sa[1000005],sb[1000005];
int to[(1<<20)+5];
int n,m,lim=1,l;
void FFT(cp *a,int len,int k)
{
    for(int i=0;i<len;i++)if(i<to[i])swap(a[i],a[to[i]]);
    for(int i=1;i<len;i<<=1)
    {
        cp w0=(cp){cos(pi/i),k*sin(pi/i)};
        for(int j=0;j<len;j+=(i<<1))
        {
            cp w=(cp){1,0};
            for(int o=0;o<i;o++,w=w*w0)
            {
                cp w1=a[j+o],w2=a[j+o+i]*w;
                a[j+o]=w1+w2,a[j+o+i]=w1-w2;
            }
        }
    }
}
cp a[(1<<20)+5],b[(1<<20)+5],c[(1<<20)+5];
ll f[1000005],g[1000005],w[1000005];
ll s3;
int ret[1000005];
int main()
{
    scanf("%d%d",&n,&m);
    scanf("%s%s",ssb,ssa);
    n=strlen(ssa),m=strlen(ssb);    
    for(int i=0;i<n;i++)sa[i]=(ssa[i]=='*')?0:ssa[i]-'a'+1;
    for(int i=0;i<m;i++)sb[i]=(ssb[i]=='*')?0:ssb[i]-'a'+1;
    for(int i=0;i<n;i++)a[i].x=(double)sa[i]*sa[i]*sa[i];
    for(int i=0;i<m;i++)b[i].x=(double)sb[m-i-1];
    while(lim<=2*n)lim<<=1,l++;
    for(int i=1;i<lim;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(l-1)));
    FFT(a,lim,1),FFT(b,lim,1);
    for(int i=0;i<lim;i++)c[i]=a[i]*b[i];
    FFT(c,lim,-1);
    for(int i=0;i<lim;i++)f[i]=(ll)(c[i].x/lim+0.5);
    memset(a,0,sizeof(a)),memset(b,0,sizeof(b)),memset(c,0,sizeof(c));
    for(int i=0;i<n;i++)a[i].x=(double)sa[i];
    for(int i=0;i<m;i++)b[i].x=(double)sb[m-i-1]*sb[m-i-1]*sb[m-i-1];
    FFT(a,lim,1),FFT(b,lim,1);
    for(int i=0;i<lim;i++)c[i]=a[i]*b[i];
    FFT(c,lim,-1);
    for(int i=0;i<lim;i++)g[i]=(ll)(c[i].x/lim+0.5);
    memset(a,0,sizeof(a)),memset(b,0,sizeof(b)),memset(c,0,sizeof(c));
    for(int i=0;i<n;i++)a[i].x=(double)sa[i]*sa[i];
    for(int i=0;i<m;i++)b[i].x=(double)sb[m-i-1]*sb[m-i-1];
    FFT(a,lim,1),FFT(b,lim,1);
    for(int i=0;i<lim;i++)c[i]=a[i]*b[i];
    FFT(c,lim,-1);
    for(int i=0;i<lim;i++)w[i]=2*(ll)(c[i].x/lim+0.5);
    int ans=0;
    for(int i=0;i+m-1<n;i++)if(f[i+m-1]+g[i+m-1]-w[i+m-1]==0)ans++,ret[ans]=i+1;
    printf("%d\n",ans);
    for(int i=1;i<=ans;i++)printf("%d ",ret[i]);
    printf("\n");
    return 0;
}

 

posted @ 2019-05-04 20:40  lleozhang  Views(161)  Comments(0Edit  收藏  举报
levels of contents