ZOJ3856 Goldbach(FFT)

//-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3856

ZOJ - 3856

参考的题解

http://www.cnblogs.com/Rojo/p/4673827.html

题意：给定n，n可以由三个及三个以内的素数通过+,*连接得到。求方案数。6=2*3与6=3*2是同一种方案。

题解：

预处理之后O(1)求解。

所有的方案有六种形式。
1，p1。预处理出素数的时候就可解。

2，p1*p2。暴力素数。

3，p1*p2*p3。暴力素数。

4，p1+p2。FFT。

5，p1+p2+p3。FFT。

6，p1*p2+p3。FFT。

FFT类的问题主要就是建模的问题。

要搞清楚指数和系数的意义。

对于计数问题，搞清楚是组合计数还是排列计数很重要。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <vector>
#include <cstdio>
#include <string>
#include <cmath>
#include <queue>
#include <set>
#include <map>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define de(x) cout << #x << "=" << x << endl

const double PI=acos(-1);
struct C {
	double x,y;
	C(double _x=0,double _y=0) {
		x=_x;y=_y;
	}
	C operator+(const C &p) {
		return C(x+p.x,y+p.y);
	}
	C operator-(const C &p) {
		return C(x-p.x,y-p.y);
	}
	C operator*(const C &p) {
		return C(x*p.x-y*p.y,x*p.y+y*p.x);
	}
};
void fft(C x[],int len,int on) {
	for(int i=1,j=len/2;j<len-1;++i) {
		if(i<j) swap(x[i],x[j]);
		int k=len>>1;
		while(j>=k) {
			j-=k;
			k>>=1;
		}
		if(j<k) j+=k;
	}
	for(int h=2;h<=len;h<<=1) {
		C wn(cos(-on*2*PI/h),sin(-on*2*PI/h));
		for(int j=0;j<len;j+=h) {
			C w(1,0);
			for(int k=j;k<j+h/2;++k) {
				C u=x[k];
				C t=w*x[k+h/2];
				x[k]=u+t;
				x[k+h/2]=u-t;
				w=w*wn;
			}
		}
	}
	if(on==-1) {
		for(int i=0;i<len;++i) x[i].x/=len;
	}
}

const int mod=1000000007;
const int N=80005;

bool vis[N];
ll p[N],cntP;
C x1[N<<2],x2[N<<2];
ll s2[N],s3[N],s4[N],s5[N],s6[N];
/*  
 *	s1:a
 *	s2:a*b
 *	s3:a*b*c
 *	s4:a+b
 *	s5:a+b+c
 *	s6:a*b+c
 */

void init() {
	memset(vis,1,sizeof vis);
	vis[0]=vis[1]=0;cntP=0;
	for(int i=2;i<N;++i) {
		if(vis[i]) {
			p[++cntP]=i;
			for(int j=2;i*j<N;++j) vis[i*j]=0;
		}
	}
	
	for(int i=1;i<=cntP&&p[i]*p[i]<N;++i) {
		for(int j=i;j<=cntP&&p[i]*p[j]<N;++j) {
			++s2[p[i]*p[j]];
			for(int k=j;k<=cntP&&p[i]*p[j]*p[k]<N;++k) {
				++s3[p[i]*p[j]*p[k]];
			}
		}
	}
	
	int len=1;
	while(len<N*2) len<<=1;
	for(int i=0;i<N;++i) x1[i]=C(vis[i],0);
	for(int i=N;i<len;++i) x1[i]=C(0,0);
	fft(x1,len,1);
	for(int i=0;i<len;++i) x1[i]=x1[i]*x1[i];
	fft(x1,len,-1);
	for(int i=0;i<N;++i) s4[i]=(ll)(x1[i].x+0.5);
	for(int i=1;i<=cntP&&p[i]*2<N;++i) --s4[p[i]*2];
	for(int i=0;i<N;++i) s4[i]>>=1;
	
	for(int i=0;i<N;++i) x1[i]=C(s4[i],0);
	for(int i=0;i<N;++i) x2[i]=C(vis[i],0);
	for(int i=N;i<len;++i) x1[i]=x2[i]=C(0,0);
	fft(x1,len,1);
	fft(x2,len,1);
	for(int i=0;i<len;++i) x1[i]=x1[i]*x2[i];
	fft(x1,len,-1);
	for(int i=0;i<N;++i) s5[i]=(ll)(x1[i].x+0.5);
	//只有两个不相同的 
	for(int i=1;i<=cntP;++i) {
		for(int j=1;j<=cntP&&p[i]*2+p[j]<N;++j) {
			if(j==i) continue;
			s5[p[i]*2+p[j]]+=2;
		}
	}
	for(int i=0;i<N;++i) s5[i]/=3; 
	
	for(int i=1;i<=cntP&&p[i]*2<N;++i) ++s4[p[i]*2];
	for(int i=1;i<=cntP&&p[i]*3<N;++i) ++s5[p[i]*3];
	
	for(int i=0;i<N;++i) x1[i]=C(s2[i],0);
	for(int i=0;i<N;++i) x2[i]=C(vis[i],0);
	for(int i=N;i<len;++i) x1[i]=x2[i]=C(0,0);
	fft(x1,len,1);
	fft(x2,len,1);
	for(int i=0;i<len;++i) x1[i]=x1[i]*x2[i];
	fft(x1,len,-1);
	for(int i=0;i<N;++i) s6[i]=(ll)(x1[i].x+0.5);
}

int main() {
	init();
	int n;
	while(~scanf("%d",&n)) {
		printf("%lld\n",vis[n]+s2[n]+s3[n]+s4[n]+s5[n]+s6[n]);
	}
    return 0;
}