砝码称重——很有意思的dp
题目传送门
题目描述
设有1g、2g、3g、5g、10g、20g的砝码各若干枚(其总重≤1000 ),
输入方式:a1,a2,a3,a4,a5,a6
(表示1g砝码有a1个,2g砝码有a2个,…,20g砝码a6个)
输出方式:Total=N
(N表示用这些砝码能称出的不同重量的个数,但不包括一个砝码也不用的情况)
思路:
1.把题目转换成01背包,把每次进来的物品个数一个个的拆分为单个物品存放在c数组内
2.将c数组进行01背包求方案数的方法求解,求解方法如下:
设f[i][j]为体积为i的方案总和
那么,当第i件物品不选时,f[i][j]=f[i-1][j]
那么如果选择了呢,f[i][j]=f[i-c[i]];
那么两者累加,就是f[i][j]=f[i-1][j]+f[i-c[i]]
为了减少空间浪费,去掉第一维,可以得到:
f[i]=f[i]+f[i-c[i]],也就是f[i]+=f[i-c[i]]
AC CODE
#include<bits/stdc++.h>
using namespace std;
const int maxn=100000;
int f[maxn]={};
int c[maxn]={};
int sum,ans;
int n;
int num[10]={0,1,2,3,5,10,20};
int main()
{
for (int i=1;i<=6;i++)
{
int k;
scanf("%d",&k);
for (int j=1;j<=k;j++) c[++n]=num[i];
}//转换01背包
f[0]=1;
for (int i=1;i<=n;i++)
sum+=c[i];//求出01背包体积最大值
for (int i=1;i<=n;i++)
for (int j=sum;j>=c[i];j--)
f[j]+=f[j-c[i]];
for (int i=1;i<=sum;i++)
if (f[i]) ans++;//累加所有可能出现的方案总和
cout<<"Total="<<ans;
return 0;
}