1423. 可获得的最大点数

最新推荐文章于 2023-12-06 00:15:00 发布

chnhbhndchngn

最新推荐文章于 2023-12-06 00:15:00 发布

阅读量108

点赞数

分类专栏： LeetCode 文章标签： python leetcode 子数组

本文链接：https://blog.csdn.net/a857553315/article/details/113731585

版权

LeetCode 专栏收录该内容

30 篇文章 0 订阅

订阅专栏

这篇博客讨论了一个中等难度的算法问题——如何在给定数组中选择k个元素以获得最大点数。作者提供了三个不同的解决方案，分别针对数组长度等于k、小于k的情况。第一版代码直接计算所有可能的k个元素组合，但超出了时间限制。第二版代码通过维护一个窗口，减少了计算次数。第三版代码考虑了特殊情况，即数组长度等于k时直接求和。博客探讨了算法优化和效率提升，适合对算法感兴趣的读者。

摘要由CSDN通过智能技术生成

1423. 可获得的最大点数

难度中等

几张卡牌 排成一行，每张卡牌都有一个对应的点数。点数由整数数组 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
解释：你必须拿起所有卡牌，可以获得的点数为所有卡牌的点数之和。

第一版代码（先取最后的K个数作为最初的结果，然后每次向后移动一位，同时从最前面取一个数字，依次进行，直到取的是最前面得K个数字，其中最大的数组和就是结果）

class Solution:
    def maxScore(self, cardPoints: List[int], k: int) -> int:
        res = sum(cardPoints[-k:])
        for i in range(1, k + 1):
            if i == k:
                li = cardPoints[:i]
            else:
                li = cardPoints[:i] + cardPoints[-(k-i):]
            tem = sum(li)
            if tem > res:
                res = tem
        return res

结果直接超出了时间限制，因为每次都在求K个元素的和，比较耗时

第二版代码（可以简化为取数组最后的K个数字，和取数据最前的K个数字组成新的数组，从中选取最大的K的子串，这次每次移动的时候就是减一个数字，加一个数字，不用每次都求和）

class Solution:
    def maxScore(self, cardPoints: List[int], k: int) -> int:
        li = cardPoints[-k:] + cardPoints[:k]
        res = temp = sum(li[:k])
        left, right = 1, k
        while right < 2*k:
            temp = temp - li[left - 1] + li[right]
            if temp > res:
                res = temp
            right += 1
            left += 1
        return res

结果还不错

第三版代码（我们再考虑一下特殊情况，就是当数组的长度就是K的时候，我们直接求和即可）

class Solution:
    def maxScore(self, cardPoints: List[int], k: int) -> int:
        if k == len(cardPoints):
            return sum(cardPoints)
            
        li = cardPoints[-k:] + cardPoints[:k]
        res = temp = sum(li[:k])
        left, right = 1, k
        while right < 2 * k:
            temp = temp - li[left - 1] + li[right]
            if temp > res:
                res = temp
            right += 1
            left += 1
        return res