leetcode 638. 大礼包

题目描述

在LeetCode商店中, 有许多在售的物品。
然而,也有一些大礼包,每个大礼包以优惠的价格捆绑销售一组物品。
现给定每个物品的价格,每个大礼包包含物品的清单,以及待购物品清单。请输出确切完成待购清单的最低花费。
每个大礼包的由一个数组中的一组数据描述,最后一个数字代表大礼包的价格,其他数字分别表示内含的其他种类物品的数量。
任意大礼包可无限次购买。

示例 1:
输入: [2,5], [[3,0,5],[1,2,10]], [3,2]
输出: 14
解释:
有A和B两种物品,价格分别为¥2和¥5。
大礼包1,你可以以¥5的价格购买3A和0B。
大礼包2, 你可以以¥10的价格购买1A和2B。
你需要购买3个A和2个B, 所以你付了¥10购买了1A和2B(大礼包2),以及¥4购买2A。

示例 2:
输入: [2,3,4], [[1,1,0,4],[2,2,1,9]], [1,2,1]
输出: 11
解释:
A,B,C的价格分别为¥2,¥3,¥4.
你可以用¥4购买1A和1B,也可以用¥9购买2A,2B和1C。
你需要买1A,2B和1C,所以你付了¥4买了1A和1B(大礼包1),以及¥3购买1B, ¥4购买1C。
你不可以购买超出待购清单的物品,尽管购买大礼包2更加便宜。

说明:

  1. 最多6种物品, 100种大礼包。
  2. 每种物品,你最多只需要购买6个。
  3. 你不可以购买超出待购清单的物品,即使更便宜。

题解

迭代,如果某个礼包special[i]是可以合法购买的(礼包物品的数量小于预购物品数),则以该礼包的价格special[i][len(needs)]加上购买剩余商品的最低价格,即对remainNeeds进行迭代

示例代码(go)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
func shoppingOffers(price []int, special [][]int, needs []int) int {
minPrice := 0
for i := 0; i < len(price); i++ {
minPrice += price[i] * needs[i]
}
for i := 0; i < len(special); i++ {
isValid := true
remainNeeds := make([]int, len(needs))
for j := 0; j < len(needs); j++ {
if special[i][j] > needs[j] {
isValid = false
break
}
remainNeeds[j] = needs[j] - special[i][j]
}
if isValid {
price := shoppingOffers(price, special, remainNeeds) + special[i][len(needs)]
if price < minPrice {
minPrice = price
}
}
}
return minPrice
}