P4173 残缺的字符串(FFT)

[Luogu4173]

题解

\(1.\)定义匹配函数

\(2.\)定义完全匹配函数

\(3.\)快速计算每一位的完全匹配函数值

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#define debug(...) fprintf(stderr,__VA_ARGS__)
#define Debug(x) cout<<#x<<"="<<x<<endl
using namespace std;
typedef long long LL;
const int INF=1e9+7;
inline LL read(){
    register LL x=0,f=1;register char c=getchar();
    while(c<48||c>57){if(c=='-')f=-1;c=getchar();}
    while(c>=48&&c<=57)x=(x<<3)+(x<<1)+(c&15),c=getchar();
    return f*x;
}
inline char readc(){
    register char c=getchar();
    while(c==' '||c=='\n'||c=='\t') c=getchar();
    return c;
}

const int MAXN=2e6+5;
const double Pi=acos(-1);

struct cmpx{
    double x,y;
    inline cmpx(){}
    inline cmpx(double _x,double _y){x=_x,y=_y;}
    inline friend cmpx operator + (cmpx a,cmpx b){return cmpx(a.x+b.x,a.y+b.y);}
    inline friend cmpx operator - (cmpx a,cmpx b){return cmpx(a.x-b.x,a.y-b.y);}
    inline friend cmpx operator * (cmpx a,cmpx b){return cmpx(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}
    inline friend cmpx operator * (cmpx a,double b){return cmpx(a.x*b,a.y*b);}
}A[MAXN],B[MAXN],C[MAXN];

char s[MAXN],t[MAXN];
int ans[MAXN],a[MAXN],b[MAXN];
int n,m;

namespace F_F_T{
    int rev[MAXN],limit,l;
    inline void init(int n){
        for(limit=1,l=0;limit<=n;limit<<=1) l++;
        for(int i=0;i<limit;i++)
            rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
    }
    inline void FFT(cmpx *A,int type){
        for(int i=0;i<limit;i++)
            if(i<rev[i]) swap(A[i],A[rev[i]]);
        for(int len=1;len<limit;len<<=1){
            cmpx Wn=(cmpx){cos(Pi/len),type*sin(Pi/len)};
            for(int i=0;i<limit;i+=(len<<1)){
                cmpx w=(cmpx){1,0};
                for(int j=0;j<len;j++,w=w*Wn){
                    cmpx x=A[i+j],y=w*A[i+len+j];
                    A[i+j]=x+y;
                    A[i+len+j]=x-y;
                }
            }
        }
        if(type==-1){
            for(int i=0;i<limit;i++) A[i].x/=limit;
        }
    }
}using namespace F_F_T;

int main(){
    n=read(),m=read();
    init(n+m-2);
    for(int i=n-1;i>=0;i--){
        char c=readc();
        a[i]=(c!='*')?c-'a'+1:0;
    }
    for(int i=0;i<m;i++){
        char c=readc();
        b[i]=(c!='*')?c-'a'+1:0;
    }

    for(int i=0;i<n;i++) A[i]=cmpx(a[i]*a[i]*a[i],0);
    for(int i=0;i<m;i++) B[i]=cmpx(b[i],0);
    FFT(A,1);FFT(B,1);
    for(int i=0;i<limit;i++) C[i]=A[i]*B[i];

    for(int i=0;i<limit;i++) A[i]=B[i]=cmpx(0,0);
    for(int i=0;i<n;i++) A[i]=cmpx(a[i],0);
    for(int i=0;i<m;i++) B[i]=cmpx(b[i]*b[i]*b[i],0);
    FFT(A,1);FFT(B,1);
    for(int i=0;i<limit;i++) C[i]=C[i]+A[i]*B[i];

    for(int i=0;i<limit;i++) A[i]=B[i]=cmpx(0,0);
    for(int i=0;i<n;i++) A[i]=cmpx(a[i]*a[i],0);
    for(int i=0;i<m;i++) B[i]=cmpx(b[i]*b[i],0);
    FFT(A,1);FFT(B,1);
    for(int i=0;i<limit;i++) C[i]=C[i]-A[i]*B[i]*2.0;

    FFT(C,-1);

    for(int i=n-1;i<m;i++){
        if(fabs(C[i].x)<0.5) ans[++ans[0]]=i-n+2;
        //printf("%.3lf\n",C[i].x);
    }
    printf("%d\n",ans[0]);
    for(int i=1;i<=ans[0];i++)
        printf("%d ",ans[i]);
}

转载于:https://www.cnblogs.com/lizehon/p/10504826.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯智能台灯

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值