砝码称重问题【dp】

设有 1g ， 2g ， 3g ， 5g ， 10g ， 20g 的砝码各若干枚 ( 其 总重≤

1000g) ，要 求：

  

输入：

       a1   a2   a3   a4   a5   a6( 表示 1g 砝码有 a1 个， 2g 砝码有 a2

个， ......20g 砝码有 a6 个 )

 

输出：

       Total=N (N 表示用这些砝码能称出的不同重量的个数，但不包括

一个砝码也不用的情况 )

 

   输入样例： 1  1  0   0   0   0

   输出样例： Total=3 ，表示可以称出 1g ， 2g ， 3g 三种不同的重

量。

code:

#include<bits/stdc++.h>
using namespace std;

int main()
{
int dp[10005],n,s[7],num,sum=0,i,j,z,ans[1005];
int m[]={1,2,3,5,10,20};
cout<<"请输入1g,2g,3g,5g,10g,20g，每种砝码的个数"<<endl;
for(i=0;i<6;i++)
{
cin>>s[i];
sum+=s[i]*m[i];
}
memset(dp,0,sizeof(dp));
dp[0]=1;
dp[sum]=1;
for(i=0;i<6;i++)
for(j=0;j<s[i];j++)
for(z=sum;z>=m[i];z--)
if(dp[z-m[i]]==1)
dp[z]=1;
for(i=1,j=0;i<=sum;i++)
if(dp[i]==1)
{
num++;
ans[j++]=i;
}
cout<<"Total="<<num<<",表示可以称出 ";
for(i=0;i<j-1;i++){
cout<<ans[i]<<"g,";
}
cout<<ans[j-1]<<"g "<<j<<"种不同的重量。"<<endl;
return 0;
}