蓝桥杯——砝码称重（DP求解）

问题描述

你有一架天平和 N个砝码，这 N个砝码重量依次是 W1, W2, · · · , Wn。

请你计算一共可以称出多少种不同的重量？ 注意砝码可以放在天平两边。

输入格式

输入的第一行包含一个整数 N。

第二行包含 N个整数：W1​,W2​,W3​,⋅⋅⋅,WN​。

输出格式

输出一个整数代表答案。

样例输入

3
1 4 6

样例输出

10

样例说明

能称出的 10 种重量是：1、2、3、4、5、6、7、9、10、11​。

1 = 1；

2 = 6 − 4 (2=6−4(天平一边放 6，另一边放 4)）；

3 = 4 − 1；

4 = 4；

5 = 6 − 1；

6 = 6；

7 = 1 + 6；

9 = 4 + 6 − 1；

10 = 4 + 6；

11 = 1 + 4 + 6；

评测用例规模与约定

对于 50%50的评测用例，1 ≤ N ≤ 15。

对于所有评测用例，1 ≤ N ≤ 100,N​个砝码总重不超过 100000。

#include<iostream>
#include<algorithm>
#include<cmath>

using namespace std;

const int  maxn=200;
long long a[maxn],n,sum=0,tot=0;
long long dp[maxn][200000];//dp[i][j]表示用到前i个砝码，能否称出j重量,,1为可以，0为不可以

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		sum+=a[i];
	}
	sort(a,a+n);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=sum;j++)
		{
			dp[i][j]=dp[i-1][j];//前i-1个砝码能称出j的重量，前i个必定也能称出j的重量 
			//每枚举新的一个砝码，都需继承前i-1的情况 
			if(dp[i][j]==0)//如果dp[i][j]==1就没有必要进行判断,j的重量由上一步继承i-1可知 
			{
        	if(j==a[i]) 
			dp[i][j]=1;//如果需要的重量正好就是第i个砝码，那么可以
        	if(dp[i-1][j+a[i]]==1) 
			dp[i][j]=1;//如果前i-1个能搞出j+a[i]重量，那么把第i个砝码放到物品一侧就行
        	if(dp[i-1][abs(j-a[i])]==1) 
			dp[i][j]=1;//如果前i-1个砝码能搞出abs(j-a[i])重量,那么把第i个砝码放天平就行
			}
		 } 
	}
	for(int j=1;j<=sum;j++)
	{
    if(dp[n][j]==1) 
	tot++;
	}
	cout<<tot;
	return 0;
 }