BZOJ 3771: Triple 母函数 FFT

3771: Triple

Time Limit: 20 Sec  Memory Limit: 64 MB
Submit: 657  Solved: 368
[Submit][Status][Discuss]

Description

我们讲一个悲伤的故事。

从前有一个贫穷的樵夫在河边砍柴。

这时候河里出现了一个水神，夺过了他的斧头，说：

“这把斧头，是不是你的？”

樵夫一看：“是啊是啊！”

水神把斧头扔在一边，又拿起一个东西问：

“这把斧头，是不是你的？”

樵夫看不清楚，但又怕真的是自己的斧头，只好又答：“是啊是啊！”

水神又把手上的东西扔在一边，拿起第三个东西问：

“这把斧头，是不是你的？”

樵夫还是看不清楚，但是他觉得再这样下去他就没法砍柴了。

于是他又一次答：“是啊是啊！真的是！”

水神看着他，哈哈大笑道：

“你看看你现在的样子，真是丑陋！”

之后就消失了。

樵夫觉得很坑爹，他今天不仅没有砍到柴，还丢了一把斧头给那个水神。

于是他准备回家换一把斧头。

回家之后他才发现真正坑爹的事情才刚开始。

水神拿着的的确是他的斧头。

但是不一定是他拿出去的那把，还有可能是水神不知道怎么偷偷从他家里拿走的。

换句话说，水神可能拿走了他的一把，两把或者三把斧头。

樵夫觉得今天真是倒霉透了，但不管怎么样日子还得过。

他想统计他的损失。

樵夫的每一把斧头都有一个价值，不同斧头的价值不同。总损失就是丢掉的斧头价值和。

他想对于每个可能的总损失，计算有几种可能的方案。

注意：如果水神拿走了两把斧头a和b，(a,b)和(b,a)视为一种方案。拿走三把斧头时，(a,b,c),(b,c,a),(c,a,b),(c,b,a),(b,a,c),(a,c,b)视为一种方案。

Input

第一行是整数N，表示有N把斧头。

接下来n行升序输入N个数字Ai，表示每把斧头的价值。

Output

若干行，按升序对于所有可能的总损失输出一行x y，x为损失值，y为方案数。

Sample Input

4
4
5
6
7

Sample Output

4 1
5 1
6 1
7 1
9 1
10 1
11 2
12 1
13 1
15 1
16 1
17 1
18 1
样例解释
11有两种方案是4+7和5+6，其他损失值都有唯一方案，例如4=4,5=5,10=4+6,18=5+6+7.

HINT

所有数据满足：Ai<=40000

为我Z基打call

快来%Z基

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>

using namespace std;

typedef long long ll;
typedef double db;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=200100;

#define pi acos(-1)

struct cp
{
	db real,image;
	friend cp operator -(const cp &x,const cp &y){return (cp){x.real-y.real,x.image-y.image};}
	friend cp operator +(const cp &x,const cp &y){return (cp){x.real+y.real,x.image+y.image};}
	friend cp operator *(const cp &x,const cp &y){return (cp){x.real*y.real-x.image*y.image,x.real*y.image+x.image*y.real};}
}a[N],b[N],c[N],f[N];

int r[N];

void fft(cp x[],int n,int p)
{
	register int i,j,k,m;
	for(i=0;i<n;++i)if(r[i]<i)swap(x[i],x[r[i]]);
	for(m=2;m<=n;m<<=1)
	{
		cp wn=(cp){cos(2*pi*p/m),sin(2*pi*p/m)};
		for(i=0;i<n;i+=m)
		{
			cp tmp,w=(cp){1,0};k=m>>1;
			for(j=0;j<k;++j,w=w*wn)
			{
				tmp=x[i+j+k]*w;
				x[i+j+k]=x[i+j]-tmp;
				x[i+j]=x[i+j]+tmp;
			}
		}
	}
}

int tmp[N],val[N],ans[N];

int main()
{
	int n=read(),maxn=0;
	register int i,j;
	for(i=1;i<=n;++i)val[i]=read(),maxn=max(maxn,val[i]);
	maxn=maxn*3/2;
	for(i=1,j=maxn;(i>>2)<j;i<<=1)maxn=i;
	for(i=0;i<maxn;++i)r[i]=((i&1)*(maxn>>1))+(r[i>>1]>>1);
	for(i=1;i<=n;++i){a[val[i]].real++;b[val[i]<<1].real++;c[val[i]*3].real++;}
	fft(a,maxn,1);fft(b,maxn,1);fft(c,maxn,1);
	for(i=0;i<maxn;++i)f[i]=a[i]+(a[i]*a[i]-b[i])*(cp){0.5,0}+(a[i]*a[i]*a[i]-(cp){3,0}*a[i]*b[i]+(cp){2,0}*c[i])*(cp){1.0/6.0,0};
	fft(f,maxn,-1);
	for(i=0;i<maxn;++i)ans[i]=f[i].real/maxn+0.1;
	for(i=0;i<maxn;++i)if(ans[i])print(i),putchar(' '),print(ans[i]),puts("");
	return 0;
}
/*
4
4
5
6
7

4 1
5 1
6 1
7 1
9 1
10 1
11 2
12 1
13 1
15 1
16 1
17 1
18 1
*/