Keshawn_lu's Blog

Leetcode 1774. 最接近目标价格的甜点成本

字数统计: 627阅读时长: 2 min
2022/12/04 Share

题目简介:

你打算做甜点,现在需要购买配料。目前共有 n 种冰激凌基料和 m 种配料可供选购。而制作甜点需要遵循以下几条规则:

  • 必须选择 一种 冰激凌基料。
  • 可以添加 一种或多种 配料,也可以不添加任何配料。
  • 每种类型的配料 最多两份

给你以下三个输入:

  • baseCosts ,一个长度为 n 的整数数组,其中每个 baseCosts[i] 表示第 i 种冰激凌基料的价格。
  • toppingCosts,一个长度为 m 的整数数组,其中每个 toppingCosts[i] 表示 一份i 种冰激凌配料的价格。
  • target ,一个整数,表示你制作甜点的目标价格。

你希望自己做的甜点总成本尽可能接近目标价格 target

返回最接近 target 的甜点成本。如果有多种方案,返回 成本相对较低 的一种。

示例 1:

1
2
3
4
5
6
7
输入:baseCosts = [1,7], toppingCosts = [3,4], target = 10
输出:10
解释:考虑下面的方案组合(所有下标均从 0 开始):
- 选择 1 号基料:成本 7
- 选择 1 份 0 号配料:成本 1 x 3 = 3
- 选择 0 份 1 号配料:成本 0 x 4 = 0
总成本:7 + 3 + 0 = 10 。

示例 2:

1
2
3
4
5
6
7
8
输入:baseCosts = [2,3], toppingCosts = [4,5,100], target = 18
输出:17
解释:考虑下面的方案组合(所有下标均从 0 开始):
- 选择 1 号基料:成本 3
- 选择 1 份 0 号配料:成本 1 x 4 = 4
- 选择 2 份 1 号配料:成本 2 x 5 = 10
- 选择 0 份 2 号配料:成本 0 x 100 = 0
总成本:3 + 4 + 10 + 0 = 17 。不存在总成本为 18 的甜点制作方案。

提示:

  • n == baseCosts.length
  • m == toppingCosts.length
  • 1 <= n, m <= 10
  • 1 <= baseCosts[i], toppingCosts[i] <= 10^4
  • 1 <= target <= 10^4

思路:

回溯,由于每种配料最多只能选两种,因此每次回溯两种状态即可。

abs(target - now_price) > abs(target - res) && now_price > target时,直接return剪枝即可。

代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
class Solution {
public:

int res = -100000;

void solve(vector<int>& toppingCosts, int now_price, int target, int pos){

if(abs(target - now_price) < abs(target - res))
res = now_price;

if(abs(target - now_price) == abs(target - res))
res = min(now_price, res);

if(now_price > target)
return;

for(int i = pos; i < toppingCosts.size(); i++){

solve(toppingCosts, now_price + toppingCosts[i], target, i + 1);
solve(toppingCosts, now_price + 2 * toppingCosts[i], target, i + 1);
}
}

int closestCost(vector<int>& baseCosts, vector<int>& toppingCosts, int target) {

for(int i = 0; i < baseCosts.size(); i++){

solve(toppingCosts, baseCosts[i], target, 0);
}

return res;
}
};
CATALOG
  1. 1. 题目简介:
  2. 2. 思路:
  3. 3. 代码如下: