最小邮票数与01背包(分别用二维数组和一维数组实现)
题目链接:最小邮票数
题目描述
有若干张邮票,要求从中选取最少的邮票张数凑成一个给定的总值。 如,有1分,3分,3分,3分,4分五张邮票,要求凑成10分,则使用3张邮票:3分、3分、4分即可。
输入描述:
有多组数据,对于每组数据,首先是要求凑成的邮票总值M,M<100。然后是一个数N,N〈20,表示有N张邮票。接下来是N个正整数,分别表示这N张邮票的面值,且以升序排列。
输出描述:
对于每组数据,能够凑成总值M的最少邮票张数。若无解,输出0。
示例1
输入
10
5
1 3 3 3 4
输出
3
思路:用01背包问题来求解。把邮票总值M当做背包容量,N张邮票的面值当成N件物品的容量,每张邮票的价值当成1,求最少邮票张数即求用N件物品塞满背包得到的最小价值。
dp[i][j]表示用前i件物品塞满容量为j的背包获得的最小价值。
dp[i][j]=min(dp[i-1][j],dp[i-1][j-w[i]]+v[i])
其中,w[i]和v[i]表示第i件物品的容量和价值,dp[i-1][j]表示第i件物品没有放入背包,只用前i-1件物品来塞满背包,dp[i-1][j-w[i]]+v[i]表示第i件物品放入背包。
分别给出用二维数组和一维数组实现的代码。
二维数组实现
#include<bits/stdc++.h>
using namespace std;
#define inf 999999
int main(){
int dp[20][100];
for(int i=0;i<20;i++){//初始化
for(int j=0;j<100;j++){
if(j==0){
dp[i][i]=0;
}
else{
dp[i][j]=inf;
}
}
}
int v[20];
for(int i=0;i<20;i++){
v[i]=1;
}
int M,N;
while(cin>>M){
cin>>N;
int w[20];
for(int i=1;i<=N;i++){
cin>>w[i];
}
for(int i=1;i<=N;i++){
for(int j=1;j<=M;j++){
if(j<w[i]){
dp[i][j]=dp[i-1][j];
}
else{
dp[i][j]=min(dp[i-1][j],dp[i-1][j-w[i]]+v[i]);
}
}
}
if(dp[N][M]==inf){
cout<<0<<endl;
}
else{
cout<<dp[N][M]<<endl;
}
}
return 0;
}
可进行空间优化,用一维数组来实现
#include<bits/stdc++.h>
using namespace std;
#define inf 999999
int main(){
int dp[100];
dp[0]=0;
for(int i=1;i<100;i++){//初始化
dp[i]=inf;
}
int v[20];
for(int i=0;i<20;i++){
v[i]=1;
}
int M,N;
while(cin>>M){
cin>>N;
int w[20];
for(int i=1;i<=N;i++){
cin>>w[i];
}
for(int i=1;i<=N;i++){
for(int j=M;j>=w[i];j--){
dp[j]=min(dp[j],dp[j-w[i]]+v[i]);
}
}
if(dp[M]==inf){
cout<<0<<endl;
}
else{
cout<<dp[M]<<endl;
}
}
return 0;
}