洛谷——P2347 砝码称重

https://www.luogu.org/problem/show?pid=2347#sub

题目描述

设有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 <algorithm>
#include <iostream>
#include <cstdio>

using namespace std;

int sum,ans;
bool f[1005];
int num[7],w[7]={0,1,2,3,5,10,20};

int main()
{
    for(int i=1;i<=6;i++)
        scanf("%d",&num[i]), sum+=num[i]*w[i];
    f[0]=1;
    for(int i=1;i<=6;i++)
        for(int j=1;j<=num[i];j++)
            for(int k=sum;k>=w[i];k--)
                if(f[k-w[i]]) f[k]=1;
    for(int i=1;i<=sum;i++)
        if(f[i]) ans++;
    printf("Total=%d",ans);
    return 0;
}