一张桌子上总共有 n 个硬币 栈 。每个栈有 正整数 个带面值的硬币。
每一次操作中,你可以从任意一个栈的 顶部 取出 1 个硬币,从栈中移除它,并放入你的钱包里。
给你一个列表 piles ,其中 piles[i] 是一个整数数组,分别表示第 i 个栈里 从顶到底 的硬币面值。同时给你一个正整数 k ,请你返回在 恰好 进行 k 次操作的前提下,你钱包里硬币面值之和 最大为多少 。
示例 1:
输入:piles = [[1,100,3],[7,8,9]], k = 2
输出:101
解释:
上图展示了几种选择 k 个硬币的不同方法。
我们可以得到的最大面值为 101 。
示例 2:
输入:piles = [[100],[100],[100],[100],[100],[100],[1,1,1,1,1,1,700]], k = 7
输出:706
解释:
如果我们所有硬币都从最后一个栈中取,可以得到最大面值和。
提示:
n == piles.length
1 <= n <= 1000
1 <= piles[i][j] <= 105
1 <= k <= sum(piles[i].length) <= 2000
解题思路:
1,这个题可以转化成背包问题
2,stack的前n个元素的和sum可以看成,重量为n,价值为sum的物品
3,对每一个栈,有0到n种选择,一共piles.length个栈
4,递推方程为f[j]=max(f[j],f[j-w] v)
5, 初始化情况下,一个物品也没有选,价值为0
6, 我们不断选物品,背包容量不断减少,价值越来越大,所以我们采用递减的方式迭代
7, 迭代完所有的栈,就得到最大值。
代码语言:javascript复制func maxValueOfCoins(piles [][]int, k int) int {
f := make([]int, k 1)
sumN := 0
for _, pile := range piles {
n := len(pile)
for i := 1; i < n; i {
pile[i] = pile[i-1] // pile 前缀和
}
sumN = min(sumN n, k) // 优化:j 从前 i 个栈的大小之和开始枚举(不超过 k)
for j := sumN; j > 0; j-- {
for w, v := range pile[:min(n, j)] {
f[j] = max(f[j], f[j-w-1] v) // 下标从 0 开始,物品体积为 w 1
}
}
}
return f[k]
}
func min(a, b int) int { if a > b { return b }; return a }
func max(a, b int) int { if b > a { return b }; return a }