题目描述
你打算做甜点,现在需要购买配料。目前共有 n 种冰激凌基料和 m 种配料可供选购。而制作甜点需要遵循以下几条规则:
必须选择 一种 冰激凌基料。
可以添加 一种或多种 配料,也可以不添加任何配料。
每种类型的配料 最多两份 。
给你以下三个输入:
baseCosts ,一个长度为 n 的整数数组,其中每个 baseCosts[i] 表示第 i 种冰激凌基料的价格。
toppingCosts,一个长度为 m 的整数数组,其中每个 toppingCosts[i] 表示 一份 第 i 种冰激凌配料的价格。
target ,一个整数,表示你制作甜点的目标价格。
你希望自己做的甜点总成本尽可能接近目标价格 target 。
返回最接近 target 的甜点成本。如果有多种方案,返回 成本相对较低 的一种。
思路分析
- 利用多重背包
- 对配料的所有价值组合的结果去重
- 将上述结果与基料组合,迭代最终成本
优化
- 数组不一定要开1000000,这里的2*pow(10,4)+1即可,时间、内存都提升一倍
source code
class Solution {
public:
vector<int> f= vector<int>(200001);
int closestCost(vector<int>& baseCosts, vector<int>& toppingCosts, int target) {
//calculate the all the remaining
set<int> se;
f[0]=0;
auto sum=[](vector<int>& t)->int{
int s=0;
for(auto &temp:t)
s+=temp*2;
return s;
};
int m=sum(toppingCosts);
for(int i=0;i<toppingCosts.size();i++) //注意存储的下标是从1开始的(阶段)
for(int j=1;j<=2;j++)
for(int k=m;k>=toppingCosts[i];k--)
f[k]=max(f[k],f[k-toppingCosts[i]]+toppingCosts[i]);
int ans=abs(toppingCosts[0]+baseCosts[0]-target);
int an=toppingCosts[0]+baseCosts[0];
for(int i=0;i<=m;i++) //f[i]表示当背包最大容量为i时,可以装下多少价值的东西
se.insert(f[i]);
for(auto &t:se)
for(auto &t2:baseCosts)
if(abs(t+t2-target)<ans||((abs(t+t2-target)==ans)&&t+t2<an)){
ans=abs(t+t2-target);
an=t+t2;
}
return an;
}
};