LeetCode 2218. 从栈中取出 K 个硬币的最大面值和

一、题目

1、题目描述

  一张桌子上总共有 $n$ 个硬币 栈 。每个栈有 正整数 个带面值的硬币。每一次操作中，你可以从任意一个栈的 顶部 取出 $1$ 个硬币，从栈中移除它，并放入你的钱包里。给你一个列表 piles，其中 piles[i]是一个整数数组，分别表示第 $i$ 个栈里 从顶到底 的硬币面值。同时给你一个正整数 $k$ ，请你返回在 恰好 进行 $k$ 次操作的前提下，你钱包里硬币面值之和最大为多少 。
  样例输入： piles = [[100],[100],[100],[100],[100],[100],[1,1,1,1,1,1,700]], k = 7
  样例输出： 706

2、基础框架

• C++ 版本给出的基础框架代码如下：

class Solution {
public:
    int maxValueOfCoins(vector<vector<int>>& piles, int k) {
    }
};

3、原题链接

LeetCode 2218. 从栈中取出 K 个硬币的最大面值和

二、解题报告

1、思路分析

  $(1)$ 由于每一堆都是一个栈，所以要取的硬币一定是连续的栈顶元素。
  $(2)$ 对于每一个栈，可以认为取 0 个，取 1 个，取 2 个，…… 取 $x$ 个，其中 $x$ 代表这个栈中有多少硬币。
  $(3)$ 于是这就转变成了一个分组背包问题。
  $(4)$ 对于每个栈的前 $i(i\ge 0)$ 个硬币，分别对应一个物品，它的容量为 $i+1$，价值为前 $i$ 个硬币的前缀和，所以对于每个栈，总共对应了 $x+1$ 个物品。
  $(5)$ 用 $dp[i][j]$ 代表前 $i$ 个栈组合出 $j$ 个硬币的最大值，则有状态转移方程：
$$dp[i][j]=max(dp[i][j],dp[i-1][j-cnt]+val);$$
  其中 $cnt$ 和 $val$ 分别对应了上文中物品的容量和价值。

有关分组背包问题的更多内容，可以参见 夜深人静写算法（十七）- 分组背包。

2、时间复杂度

   最坏时间复杂度 $O(nk)$ 。

3、代码详解

class Solution {
public:
    int maxValueOfCoins(vector<vector<int>>& piles, int k) {
        int dp[1010][2010];
        int i, j, x;
        int n = piles.size();
        int totK = 0;
        memset(dp, 0, sizeof(dp));
        dp[0][0] = 0;
        for(i = 1; i <= n; ++i) {
            int psize = piles[i-1].size();
            totK += psize;
            if(totK > k) {
                totK = k;
            }
            for(j = 1; j < psize; ++j) {
                piles[i-1][j] += piles[i-1][j-1];
            }
            
            for(j = totK; j >= 0; --j) {
                for(x = -1; x < psize; ++x) {
                    int val = (x == -1 ? 0 : piles[i-1][x]);
                    int cnt = (x + 1);
                    if(j < cnt) {
                        break;
                    }
                    dp[i][j] = max(dp[i][j], dp[i-1][j - cnt] + val);
                }
            }
        }
        return dp[n][k];
        
    }
};

三、本题小知识

  看着是有三层循环嵌套的DP，但是由于数据量的限制，总的时间复杂度没有想象中的那么高。

四、加群须知

  相信看我文章的大多数都是「 大学生 」，能上大学的都是「 精英 」，那么我们自然要「 精益求精 」，如果你还是「 大一 」，那么太好了，你拥有大把时间，当然你可以选择「 刷剧 」，然而，「 学好算法 」，三年后的你自然「 不能同日而语 」。
  那么这里，我整理了「 几十个基础算法 」 的分类，点击开启：

  大致题集一览：

  为了让这件事情变得有趣，以及「 照顾初学者 」，目前题目只开放最简单的算法 「 枚举系列 」 （包括：线性枚举、双指针、前缀和、二分枚举、三分枚举），当有 一半成员刷完 「 枚举系列 」 的所有题以后，会开放下个章节，等这套题全部刷完，你还在群里，那么你就会成为「 夜深人静写算法 」专家团 的一员。
  不要小看这个专家团，三年之后，你将会是别人 望尘莫及 的存在。如果要加入，可以联系我，考虑到大家都是学生， 没有「 主要经济来源 」，在你成为神的路上，「 不会索取任何 」。
  🔥联系作者，或者扫作者主页二维码加群，加入刷题行列吧🔥

🔥让天下没有难学的算法🔥
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
让你养成九天持续刷题的习惯 🔥《九日集训》🔥
入门级C语言真题汇总 🧡《C语言入门100例》🧡
组团学习，抱团生长 🌌《算法零基础100讲》🌌
几张动图学会一种数据结构 🌳《画解数据结构》🌳
竞赛选手金典图文教程 💜《夜深人静写算法》💜