【BZOJ】3771: Triple

http://www.lydsy.com/JudgeOnline/problem.php?id=3771

题意:n个带价值互不相同的物品,每次可以取1、2、3个物品,问能得到的所有的价值和这个价值的方案数(n不明(无意义= =),价值<=40000)

#include <bits/stdc++.h>
using namespace std;
const int N=200005;
int bit[N];
const double PI=acos(-1);
struct cp {
	double r, i;
	cp(double _r=0, double _i=0) : r(_r), i(_i) {}
	cp operator + (const cp &a) { return cp(r+a.r, i+a.i); }
	cp operator - (const cp &a) { return cp(r-a.r, i-a.i); }
	cp operator * (const cp &a) { return cp(r*a.r-i*a.i, r*a.i+i*a.r); }
};
void dft(cp *a, int n, int flag) {
	for(int i=0; i<n; ++i) if(i<bit[i]) swap(a[i], a[bit[i]]);
	for(int m=2; m<=n; m<<=1) {
		cp wn(cos(PI*2.0/m), sin(PI*2.0/m)*flag);
		int mid=m>>1;
		for(int i=0; i<n; i+=m) {
			cp w(1);
			for(int j=0; j<mid; ++j) {
				static cp u, v;
				u=a[i+j], v=a[i+j+mid]*w;
				a[i+j]=u+v;
				a[i+j+mid]=u-v;
				w=w*wn;
			}
		}
	}
	if(flag==-1) for(int i=0; i<n; ++i) a[i].r/=n;
}
void fft(int *A, int *B, int *C, int n) {
	static cp a[N], b[N];
	int len=1, bitl=-1;
	for(; len<n; len<<=1, ++bitl);
	for(int i=0; i<len; ++i) bit[i]=(bit[i>>1]>>1)|((i&1)<<bitl);
	for(int i=0; i<len; ++i) a[i].r=A[i], a[i].i=0, b[i].r=B[i], b[i].i=0;
	dft(a, len, 1); dft(b, len, 1);
	for(int i=0; i<len; ++i) b[i]=a[i]*b[i];
	dft(b, len, -1);
	for(int i=0; i<len; ++i) C[i]=b[i].r+0.5;
}
int a[N], b[N], c[N], t[N], ans[N], n, len;
void work() {
	int l=n;
	for(int i=0; i<l; ++i) ans[i]=a[i];
	fft(a, a, t, n*2-1);
	l=n*2-1;
	for(int i=0; i<l; ++i) ans[i]+=(t[i]-b[i])>>1;
	fft(a, t, t, n*3-2);
	fft(a, b, a, n*3-2);
	l=n*3-2;
	for(int i=0; i<l; ++i) ans[i]+=(t[i]-3*a[i]+(c[i]<<1))/6;
}
int main() {
	scanf("%d", &len);
	for(int i=0; i<len; ++i) { int x; scanf("%d", &x); a[x]=1; b[x*2]=1; c[x*3]=1; n=max(x+1, n); }
	work();
	len=n*3-2;
	for(int i=0; i<len; ++i) if(ans[i]) printf("%d %d\n", i, ans[i]);
	return 0;
}

  

首先容易得到母函数$A=\sum_{存在价值为i的物品} x^i$

敲完了fft才发现如果直接求母函数的三次方是不对的= =...

妈呀竟然没想到容斥QAQ

设$B = \{A[i]^2\}, C = \{A[i]^3\}$

首先我们对取法分别求:

取1个的方案 $ = A$

取2个的方案 $ = \frac{A^2 - B}{A^{2}_{2}} = \frac{A^2 - B}{2}$

取3个的方案 $ = \frac{A^3 - \frac{3!}{2!1!} AB + \frac{3!}{2!1!} C - C}{A^{3}_{3}} = \frac{A^3 -3AB + 2C}{6}$

(看不懂的建议先去看《组合数学》= =)

然后fft搞搞就行辣= =

时间: 2024-10-10 10:12:28

【BZOJ】3771: Triple的相关文章

【BZOJ】3319: 黑白树

http://www.lydsy.com/JudgeOnline/problem.php?id=3319 题意:给一棵n节点的树(n<=1e6),m个操作(m<=1e6),每次操作有两种:1.查询u到根的第一条黑边的编号.2.将u到v的路径全部染成黑色 #include <cstdio> #include <cstring> #include <cmath> #include <string> #include <iostream>

【bzoj】4538: [Hnoi2016]网络

题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=4538 维护一个数据结构支持对于一颗树的操作,需要支持: 1.对于树上的一条路径上的每个点上放一个值. 2.撤销某次操作的路劲放. 3.查询除了经过这个点的路径的最大值. 往一个路径上丢值相当于往不经过条路径的所有点上丢值. 用一个树链剖分即可维护,对于操作区间取反. 直接查询单点最大值即可. 为了维护单点最大值,线段树中的每一个点对应两个堆,用于维护插入誉删除. 防止爆空间,所以标记永久

【BZOJ】1821: [JSOI2010]Group 部落划分 Group(最小生成树+贪心)

http://www.lydsy.com:808/JudgeOnline/problem.php?id=1821 这题裸题. 本题要求最短距离最长,很明显,我们排序. 这里存在贪心,即我们把边权最小的全分给n个部落的内部,然后剩下的边最小的就是答案. 将边权较小的边分给k个部落,用并查集生成最小树,使得内部的边总是小于连到外部的边.然后分剩下k个点即可,剩下的k个点的那条边一定是部落之间最小的且最长的边. #include <cstdio> #include <cstring> #

【BZOJ】【3083】遥远的国度

树链剖分/dfs序 其实过了[BZOJ][4034][HAOI2015]T2以后就好搞了…… 链修改+子树查询+换根 其实静态树的换根直接树链剖分就可以搞了…… 因为其实只有一样变了:子树 如果root在x的子树中(以1为根dfs的时候),那么现在x的子树就变成了整个dfs序中,除去含有root的那个子树的剩下的部分,画个图大概就是这样:(红色部分为现在的子树) 我们发现,这种子树由于换根而产生变化的情况,仅当在以1为根时的树中,x是new_root的祖先时发生,那么我们判断这种情况是否发生只需

【BZOJ】2301: [HAOI2011]Problem b(莫比乌斯+分块)

http://www.lydsy.com/JudgeOnline/problem.php?id=2301 和这题不是差不多的嘛--[BZOJ]1101: [POI2007]Zap(莫比乌斯+分块) 唯一不同的地方是这题有下界.. 下界除以k的时候取上界,然后分块的时候因为有4个数,所以要分成4块来搞.. 然后就行了.. #include <cstdio> #include <cstring> #include <cmath> #include <string>

【BZOJ】1146: [CTSC2008]网络管理Network(树链剖分+线段树套平衡树+二分 / dfs序+树状数组+主席树)

第一种做法(时间太感人): 这题我真的逗了,调了一下午,疯狂造数据,始终找不到错. 后来发现自己sb了,更新那里没有打id,直接套上u了.我.... 调了一下午啊!一下午的时光啊!本来说好中午A掉去学习第二种做法,噗 好吧,现在第一种做法是hld+seg+bst+二分,常数巨大,log^4级别,目前只会这种. 树剖后仍然用线段树维护dfs序区间,然后在每个区间建一颗平衡树,我用treap,(这题找最大啊,,,囧,并且要注意,这里的rank是比他大的数量,so,我们在二分时判断要判断一个范围,即要

最大权闭合图 &amp;&amp; 【BZOJ】1497: [NOI2006]最大获利

最大权闭合图详细请看胡伯涛论文<最小割模型在信息学竞赛中的应用>,我在这里截图它的定义以及一些东西. 假设我们有一个图,点集的出边都是连到点集的,那么称这个为闭合图.现在这些点集都有个权值,我们要选择某个闭合图使得权值最大. 回到此题: 最大获利这一题,我们可以这样看,用户群和中转站为带权的点集,用户群的权为收益,中转站的权为负的成本,即0-成本,用户群向其中两个中转站连弧,那么这个就是一个闭合图. 我们要求这个闭合图的权值和最大,即最大收益,那么就能转移到上面的求最大权闭合图的做法去了. 做

【BZOJ】1012: [JSOI2008]最大数maxnumber(树状数组+区间最值)

http://www.lydsy.com/JudgeOnline/problem.php?id=1012 树状数组原来我只懂得sum和add的操作,今天才知道可以有求区间最值的操作,我学习了一下写了个,1a了. 区间最值其实和区间求和差不多,就是将sum数组的含义转移到max,然后通过特定的区间更新max. 在区间求和中,当我们维护max[i]的时候,要找到它前面所有的max[j]来更新,在这里因为是树状数组,所以可以降成一个log级,画图可知,max[i]需要的max只有max[i-2^0],

【BZOJ】3052: [wc2013]糖果公园

http://www.lydsy.com/JudgeOnline/problem.php?id=3052 题意:n个带颜色的点(m种),q次询问,每次询问x到y的路径上sum{w[次数]*v[颜色]},可以单点修改颜色.(n, m, q<=100000) #include <bits/stdc++.h> using namespace std; const int N=100005, M=100005; typedef long long ll; inline int getint()