1、题目描述
https://leetcode-cn.com/problems/maximum-points-you-can-obtain-from-cards/
几张卡牌 排成一行,每张卡牌都有一个对应的点数。点数由整数数组 cardPoints 给出。
每次行动,你可以从行的开头或者末尾拿一张卡牌,最终你必须正好拿 k 张卡牌。
你的点数就是你拿到手中的所有卡牌的点数之和。
给你一个整数数组 cardPoints 和整数 k,请你返回可以获得的最大点数。
输入:cardPoints = [1,2,3,4,5,6,1], k = 3
输出:12
解释:第一次行动,不管拿哪张牌,你的点数总是 1 。但是,先拿最右边的卡牌将会最大化你的可获得点数。最优策略是拿右边的三张牌,最终点数为 1 + 6 + 5 = 12 。
输入:cardPoints = [2,2,2], k = 2
输出:4
解释:无论你拿起哪两张卡牌,可获得的点数总是 4 。
2、代码详解
类似题:思路都是滑动窗口。或者前缀和
643. 子数组最大平均数 I(前缀和、滑动窗口)https://blog.csdn.net/IOT_victor/article/details/113705185
把剩余的中间部分元素抽象成长度固定为 windowSize = N - k 的滑动窗口。
当每次窗口右移的时候,需要把右边的新位置 加到 窗口中的 和 中,把左边被移除的位置从窗口的 和 中 减掉。
这个方法只用遍历一次数组。需要根据 i 的位置,计算滑动窗口是否开始、是否要移除最左边元素:
- 当 i >= windowSize 时,为了固定窗口的元素是 k 个,每次移动时需要将 i - windowSize 位置的元素移除。
- 当 i >= windowSize - 1 时,滑动窗口内的元素刚好是 k 个,开始计算滑动窗口的最小和。
- 用 cardPoints 的所有元素之和,减去滑动窗口内的最小元素和,就是拿走的卡牌的最大点数。
from typing import List
class Solution:
def maxScore(self, cardPoints: List[int], k: int) -> int:
n = len(cardPoints)
window_size = n - k
sums = 0
res = float('inf') # 滑动窗口内的最小元素和
for i in range(n):
sums += cardPoints[i]
if i >= window_size:
sums -= cardPoints[i - window_size]
if i >= window_size - 1:
res = min(res, sums)
return sum(cardPoints) - res
cardPoints = [1, 2, 3, 4, 5, 6, 1]
k = 3
s = Solution()
print(s.maxScore(cardPoints, k))