zoukankan      html  css  js  c++  java
  • BZOJ 4503: 两个串

    先Orz: 一类关于通配符匹配的字符串问题都是可以用FFT来解决的

    题面:bzoj 4503

    pre : 看了题发现这题不是和CodeForces_528D一样吗,然后就同样的写了。。26遍FFT,果断TLE

    题解

    P.S. 对于这类只有相等匹配和万能匹配的字符串问题可以一遍FFT

    (S)为文本串,(T)为模式串((m=T.size())),然后构造

    [a[i]=S[i],b[i]=(T[i]=='?'?0:T[i]) ]

    [c[i]=sum_{j=0}^{m-1}(a[i+j]-b[j])^2b[j] ]

    然后考虑到此时如果(s[i->i+m])(T)匹配当且仅当(c[i]=0)

    那么开始转变成卷积形式

    [c[i]=sum_{j=0}^{m-1}a[i+j]^2b[j]-2a[i+j]b[j]^2+b[j]^3 ]

    那么构造三个卷积(F,G,H)

    [F[i]=sum_{j=0}^{m-1}a[i+j]^2b[j] ]

    [=>F'[m+i-1]=sum_{j=0}^{m-1}a[i+j]^2b'[m-j-1] ]


    [G[i]=sum_{j=0}^{m-1}2a[i+j]b[j]^2 ]

    [=>G'[m+i-1]=sum_{j=0}^{m-1}2 imes a[i+j]b'[m-j-1]^2 ]


    [H[i]=sum_{j=0}^{m-1}b[j]^3 ]


    #include<bits/stdc++.h>
    
    using namespace std;
    
    namespace Tzh{
    	
    	typedef long double dd;
    	const int maxn=8e5+10;
    	const dd pi=acos(-1.0L);
    	int pr[maxn],cnt,m,n=1,len,rev[maxn];
    	string S,T;
    	
    	struct complex{
    		dd x,y;
    		complex operator +(const complex &b) const{
    			return (complex){x+b.x,y+b.y}; 
    		}
    		complex operator -(const complex &b) const{
    			return (complex){x-b.x,y-b.y};	
    		}
    		complex operator *(const complex &b) const{
    			return (complex){x*b.x-y*b.y,x*b.y+y*b.x};	
    		}
    		complex operator *(const dd b) const{
    			return (complex){x*b,y*b};	
    		}
    	}omg[maxn],s[maxn],inv[maxn],c[maxn],s2[maxn],f[maxn],g[maxn],
    	t[maxn],t_rev[maxn],h,t2[maxn],t2_rev[maxn];
    	
    	void init(){
    		for(int i=0;i<n;i++)
    			omg[i]=(complex){cos(i*2*pi/n),sin(i*2*pi/n)},
    			inv[i]=(complex){cos(i*2*pi/n),-sin(i*2*pi/n)};	
    		for(int i=1;i<n;i++)
    			rev[i]=(rev[i>>1]>>1)|((i&1)<<(len-1));
    	}
    	
    	void fft(complex *a,complex *omg){
    		for(int i=1;i<n;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
    		for(int l=2,m=1;l<=n;m=l,l<<=1)
    			for(int i=0;i<n;i+=l)
    				for(int j=0;j<m;j++){
    					complex tt=omg[n/l*j]*a[i+j+m];
    					a[i+j+m]=a[i+j]-tt,a[i+j]=a[i+j]+tt;
    				}
    		if(omg==inv) 
    			for(int i=0;i<n;i++) a[i].x/=n;
    	}
    	 
    	void work(){
    		ios::sync_with_stdio(false);
    		cin>>S>>T; m=T.size(); int tmp=S.size();
    		while(n<=S.size()+m) n<<=1,len++;
    		init();
    		for(int i=0;i<S.size();i++) s[i].x=S[i];
    		for(int i=0;i<m;i++) t[i].x=T[i]=='?'?0:T[i];
    		for(int i=0;i<m;i++) t_rev[i]=t[m-i-1];
    		for(int i=0;i<m;i++) t2[i]=t[i]*t[i];
    		for(int i=0;i<m;i++) t2_rev[i]=t2[m-i-1];
    		for(int i=0;i<S.size();i++) s2[i]=s[i]*s[i];
    		for(int i=0;i<m;i++) h=h+t[i]*t[i]*t[i];
    		fft(t_rev,omg);
    		fft(t2_rev,omg),fft(s,omg),fft(s2,omg);
    		for(int i=0;i<n;i++) f[i]=s2[i]*t_rev[i]; 
    		for(int i=0;i<n;i++) g[i]=s[i]*t2_rev[i];
    		for(int i=0;i<n;i++) c[i]=f[i]-g[i]*2.0; fft(c,inv);
    		for(int i=0;i<S.size()-m+1;i++) if((int)(c[m+i-1].x+0.5+h.x)==0) pr[++cnt]=i;
    		cout<<cnt<<endl;
    		for(int i=1;i<=cnt;i++) cout<<pr[i]<<endl;
    	}
    }
    
    int main(){
    	Tzh::work();
    	return 0;	
    }
    
  • 相关阅读:
    C#2.0新特性
    .NET Framework 4.5新特性
    C#4.0 中的新特性
    C#3.0中的新特性
    开始使用Mac OS X——写给Mac新人
    关于WF4 State Machine 状态机工作流的发布问题.
    C# 3.5新特性
    libssh2编译错误(configure error: cannot find OpenSSL or Libgcrypt)解决方法
    程序员转产品经理的思考
    升级PHP版本
  • 原文地址:https://www.cnblogs.com/tang666/p/9204200.html
Copyright © 2011-2022 走看看