PAT-A1068 Find More Coins
01背包问题: 原题链接.
01背包问题详解链接: 01背包.
在本题中考虑物品重量=价值
代码
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int dp[10100], w[10100]; //dp为背包大小为i时的价值和,w为每个物品的重量
bool choice[110][10100]; //choice[j][i],存放容量为j时,是否选择了i物品
int cmp1(int a, int b){ //根据题意将物品从大到小排列,
return a>b; //由背包的循环,相同背包容量下,将小重量的物品更新进去
}
int main(){
int n,m;
cin>>n>>m; //n为物品总数,m为要求的物品和
for(int i=1;i<=n;++i){
cin>>w[i];
}
sort(w+1,w+n+1,cmp1); //将物品从大到小排列
for(int i=1;i<=n;++i){ //01背包问题的格式,i为w数组从1到n,依次考虑不同物品
for(int j=m;j>=w[i];--j){ //动态规划+压缩空间,只考虑物品重量小于总容量的情况
if(dp[j] <= dp[j-w[i]]+w[i]){ //如果i物品+上一循环考虑的容量j-i重量下的最优解dp[j-w[i]]>=不放入i(即dp[j],即上一循环),则更新dp,将i放入
choice[j][i]=true; //choice[j][i]表示容量为j时,i物品是否放入
dp[j] = dp[j-w[i]]+w[i]; //更新dp
}
}
}
if(dp[m] != m){ //如果01书包问题的最高价值即本题的重量,不等于m
printf("No Solution");
}
else {
vector<int> arr;
int v = m, index = n;
while(v > 0){ //choice二维数组j为容量从小到大,i为物品,重量从大到小
if(choice[v][index]){ //将choice二维数组从容量为m开始考虑,从重量最小的物品开始考虑
arr.push_back(w[index]); //将小重量的放入数组
v -= w[index]; //考虑减掉小重量物品的情况
}
--index; //考虑下一个重量最小的物品
}
for(int i=0;i<arr.size();++i){
if(i) printf(" ");
printf("%d", arr[i]);
}
}
return 0;
}