题目是LeetCode第186场周赛的第二题,链接:可获得的最大点数。具体描述为:几张卡牌 排成一行,每张卡牌都有一个对应的点数。点数由整数数组cardPoints
给出。每次行动,你可以从行的开头或者末尾拿一张卡牌,最终你必须正好拿k
张卡牌。你的点数就是你拿到手中的所有卡牌的点数之和。给你一个整数数组cardPoints
和整数k
,请你返回可以获得的最大点数。
示例1:
输入:cardPoints = [1,2,3,4,5,6,1], k = 3
输出:12
解释:第一次行动,不管拿哪张牌,你的点数总是 1 。但是,先拿最右边的卡牌将会最大化你的可获得点数。最优策略是拿右边的三张牌,最终点数为 1 + 6 + 5 = 12 。
示例2:
输入:cardPoints = [2,2,2], k = 2
输出:4
解释:无论你拿起哪两张卡牌,可获得的点数总是 4 。
示例3:
输入:cardPoints = [9,7,7,9,7,7,9], k = 7
输出:55
解释:你必须拿起所有卡牌,可以获得的点数为所有卡牌的点数之和。
示例4:
输入:cardPoints = [1,1000,1], k = 1
输出:1
解释:你无法拿到中间那张卡牌,所以可以获得的最大点数为 1 。
示例5:
输入:cardPoints = [1,79,80,1,1,1,200,1], k = 3
输出:202
题目比较简单,直接遍历所有可能的拿牌组合即可,即(左边k张,右边0张)、(左边k-1张,右边1张)直到(左边0张,右k张)这么多种组合,取其中点数最大的即可。需要注意的是我们不需要对每个组合都重新计算点数和,而是可以充分利用相邻两个组合中只有两张不同牌的特点,利用前一个组合的点数和进行简单的加减即可得到当前组合的点数和。时间复杂度为 O ( k ) O(k) O(k),空间复杂度为 O ( 1 ) O(1) O(1)。
JAVA版代码如下:
class Solution {
public int maxScore(int[] cardPoints, int k) {
int result = 0;
for (int i = 0; i < k; ++i) {
result += cardPoints[i];
}
int curSum = result;
for (int i = 0; i < k; ++i) {
curSum = curSum - cardPoints[k - i - 1] + cardPoints[cardPoints.length - 1 - i];
result = Math.max(result, curSum);
}
return result;
}
}
提交结果如下:
Python版代码如下:
class Solution:
def maxScore(self, cardPoints: List[int], k: int) -> int:
N = len(cardPoints)
result = sum(cardPoints[0 : k])
curSum = result
for i in range(k):
curSum += cardPoints[N - 1 - i] - cardPoints[k - 1 - i]
result = max(result, curSum)
return result
提交结果如下: