天平问题

天平问题

题目

小C为了试验小X，便为物竞的小X出了一道物理相关的题：现在给出n个质量的砝码，问小X能称出多少种质量的物品，可是总有好事者想要破坏，于是乎，n达到了500，远远超出了小X能够承受的范围，锲而不舍的他决定寻求你们的帮助。

注意：天平有两边，两边均可放。

输入

第一行输入一个的正整数N
以下N行每行一个不超过200的正整数，依次表示每个砝码的质量。

输出

输出总共能称出多少种不同质量的物品。

输入样例

3
1
3
9

输出样例

13

思路

这道题我们用动态规划来做。
设f[i][j]为用前i个砝码能否称出质量为j的物品。那么我们知道，每一种物品可以放左边，也可以放右边。所以动态转移方程就是：$f[i][j+a[i]]=f[i][j-a[i]]=f[i][j]=1;$
当然，这个方程要建立在f[i-1][j]存在的情况。
最后，我们统计能称出的每一种重量，然后输出就可以了。

代码

#include<cstdio>
using namespace std;
int n,a[501],ans,tot;
bool f[501][40001];
int main()
{
	scanf("%d",&n);//读入
	for (int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);//读入
		tot+=a[i];//求出称出的值最大可能是多少
	}
	f[1][20000-a[1]]=f[1][20000+a[1]]=f[1][20000]=1;//初始化
	for (int i=2;i<=n;i++)
	for (int j=20000-tot;j<=20000+tot;j++)
	if (f[i-1][j]) f[i][j+a[i]]=f[i][j-a[i]]=f[i][j]=1;//动态转移方程
	for (int i=20001;i<=20000+tot;i++)
	if (f[n][i]) ans++;//统计能称出的重量数
	printf("%d",ans);//输出
	return 0;
}