天平问题 动态规划

天平问题

题目描述

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

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

输入格式

第一行，一个正整数N；

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

输出格式

一行，一个数，表示总共能称出多少种不同质量的物品。

输入样例

3
1
3
9

输出样例

13

---

解题思路

本题大意：求用这N个砝码可以称出不同质量物品的方案数是多少？
求方案数首选动态规划，我们先来分析一下题目：因为天平的两边都可以放砝码，所以这些砝码可以相加，也可以相减而得到不同的质量。
所以我们不妨将整个问题分成两个部分：砝码相加，砝码相减。
设$f[i]$表示$i$是否可以通过砝码的组合得到：

---

砝码相加

由于一种砝码只有一个，所以如果我们从小的数往大的数动态规划，就会出现同一种砝码累加的错误，我们从大数开始递减动态规划。

for(int j=1;j<=n;j++)
	for(int i=maxn;i>=0;i--)
		if(f[i]>0)
			f[i+a[j]]++;

---

---

砝码相减

为了避免同一种砝码多次减去的错误，我们从小到大的动态规划，其它都跟砝码相加一样。

for(int j=1;j<=n;j++)
	for(int i=0;i<=maxn;i++)
		if(f[i]>0&&i-a[j]>0)//防止下标越界
			f[i-a[j]]++;

---

最后枚举整个$f$数组，如果$f[i]>0$，方案数加1。

---

代码

#include<iostream>

using namespace std;
int n;
int a[505],ans;
int f[50005];

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	f[0]=1;//初始化
	for(int i=1;i<=n;i++)
		for(int j=20000;j>=0;j--)
			if(f[j]>0)
				f[j+a[i]]++;
	for(int i=1;i<=n;i++)
		for(int j=0;j<=20000;j++)
			if(f[j]>0)
				if(j-a[i]>0)
					f[j-a[i]]++;
	for(int i=1;i<=20000;i++)
		if(f[i]>0)
			ans++;
	cout<<ans;
	return 0;
}