洛谷——2347砝码称重

题目描述

设有1g、2g、3g、5g、10g、20g的砝码各若干枚（其总重<=1000），

输入输出格式

输入格式：

输入方式：a1 a2 a3 a4 a5 a6

（表示1g砝码有a1个，2g砝码有a2个，…，20g砝码有a6个）

输出格式：

输出方式：Total=N

（N表示用这些砝码能称出的不同重量的个数，但不包括一个砝码也不用的情况）

输入输出样例

输入样例#1：

1 1 0 0 0 0

输出样例#1：

Total=3

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
using namespace std;
int a[7],sum=0,ans=0;
int w[7]={0,1,2,3,5,10,20};
bool dp[1010];
int main()
{
    memset(dp,false,sizeof(dp));dp[0]=true;
    for(int i=1;i<=6;i++) scanf("%d",&a[i]),sum+=a[i]*w[i];
    for(int k=1;k<=6;k++)
     for(int j=1;j<=a[k];j++)
      for(int i=sum;i>=w[k];i--)
        if(dp[i-w[k]]) dp[i]=true;//不能写成j*w[k]因为这样会导致物品数量超过 想一想 为什么？
    for(int i=1;i<=sum;i++) if(dp[i]) ans++;
    printf("Total=%d",ans);
    return 0;
}