题目 2604: 蓝桥杯2021年第十二届省赛真题-砝码称重

时间限制: 1s 内存限制: 128MB 提交: 4609 解决: 1510

题目描述

你有一架天平和 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 (天平一边放 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% 的评测用例，1 ≤ N ≤ 15。
对于所有评测用例，1 ≤ N ≤ 100，N 个砝码总重不超过 100000。

求解思路：使用动态规划，dp[i][j]的值为0或1，表示前i个砝码能否组成重量j，如果能则dp[i][j]=1,否则为0；

动态转移方程为

（1）当前砝码刚好等于重量j if(j==w[i]) dp[i][j]=1

（2）前i-1个砝码能够组成重量 j+a[i] ，则组成j重量时，只需要在组成j+a[i]重量的天平下重的一方取下a[i] 砝码就可以构成 j 重量

       if(dp[i-1][j-w[i]]==1) dp[i][j]=1

（3）当前i-1个砝码能够组成abs(j-a[i])时，假设两边砝码为x和y，x-y=j-a[i],只需要在重的一方加上a[i]就可以组成重量j ,即(x+a[i])-y=j

       if(dp[i-1][Math.abs(j-w[i])]==1) dp[i][j]=1;

import java.util.*;
public class Main{
  public static void main(String args[])
  {   final int N=100000;
	  Scanner in=new Scanner(System.in);
	  int n;
	  int w[];
	  int type=0;
	  int sum=0;
	  n=in.nextInt();
	  int dp[][]=new int[n+1][N];
	  w=new int[n+1];
	  for(int i=1;i<=n;i++)
		  {w[i]=in.nextInt();Main
		   sum+=w[i];
		  }
	  //只有left有砝码
	  //Arrays.fill(temp, 0);
	  //Arrays.fill(dp,temp);
	  for(int i=1;i<=n;i++)
	  {
		  for(int j=1;j<=sum;j++)
		  {
			  dp[i][j]=dp[i-1][j];
			  if(w[i]==j) dp[i][j]=1;
			  if(dp[i-1][j+w[i]]==1) dp[i][j]=1;//减去一个砝码即可
			  if(dp[i-1][Math.abs(j-w[i])]==1) dp[i][j]=1;//增加一个砝码即可
		  }
	  }
	  
	  for(int j=1;j<=sum;j++) if(dp[n][j]==1) type++;
	  System.out.println(type);
  }
}