砝码称重(dp)

砝码称重

内容：

砝码称重（30分）

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

要求：

输入方式：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三种不同的重量。

#include <iostream>
#include <stdio.h>
#include <string>
#include <string.h>
#include <map>
#include <algorithm>
#include <math.h>
#include <vector>
#include <cstring>
#include <cstdio>
using namespace std;
bool dp[1005];
int num[7],size[7],sum;
int main(){
 num[1]=1;//虽然数组从零开始，这里赋值应该从一开始因为下面dp数组dp[0]=1,而dp[k]|=dp[k-num[i]] 
  num[2]=2;
  num[3]=3;
  num[4]=5;
  num[5]=10;
  num[6]=20;
   int i,j,k,ans;
  ans=sum=0;
 for(i=1;i<=6;i++){
  scanf("%d",&size[i]);
   sum+=num[i]*size[i];
  }
  memset(dp,0,sizeof(dp));
  dp[0]=1;
  for(i=1;i<=6;i++)
   for(j=1;j<=size[i];j++)
     for(k=sum;k>=num[i];k--)
       dp[k]|=dp[k-num[i]];
  for(i=sum;i>=1;i--){
    if(dp[i])
      ans++;
  }
 printf("Total=%d\n",ans);
  return 0;
}