LeetCode-day32-LCP 40. 心算挑战

题目描述

「力扣挑战赛」心算项目的挑战比赛中，要求选手从 N 张卡牌中选出 cnt 张卡牌，若这 cnt 张卡牌数字总和为偶数，则选手成绩「有效」且得分为 cnt 张卡牌数字总和。 给定数组 cards 和 cnt，其中 cards[i] 表示第 i 张卡牌上的数字。 请帮参赛选手计算最大的有效得分。若不存在获取有效得分的卡牌方案，则返回 0。

示例

示例1：

输入：cards = [1,2,8,9], cnt = 3
输出：18
解释：选择数字为 1、8、9 的这三张卡牌，此时可获得最大的有效得分 1+8+9=18。

示例2：

输入：cards = [3,3,1], cnt = 1
输出：0
解释：不存在获取有效得分的卡牌方案。

思路

排序+贪心

为了选取尽量大的数，将 cards 从大到小排序，并累加前 cnt 个数，记作 s。

分类讨论：

如果 s 是偶数，这是我们能得到的最大得分，直接返回 s。
如果 s 是奇数，那么可以：

• 从前 cnt 个数中去掉一个最小的奇数，从后 n−cnt 个数中加进来一个最大的偶数，这样得分就变成偶数了。
• 从前 cnt 个数中去掉一个最小的偶数，从后 n−cnt 个数中加进来一个最大的奇数，这样得分就变成偶数了。
• 两种情况取最大值。

代码

class Solution:
    def maxmiumScore(self, cards: List[int], cnt: int) -> int:
        cards.sort(reverse=True)
        s = sum(cards[:cnt])
        if s % 2 == 0:
            return s

        def replaced_sum(x:int) -> int:
            for v in cards[cnt:]:
                if v % 2 != x %2:
                    return s-x +v
            return 0

        x = cards[cnt-1]
        ans = replaced_sum(x)
        for v in cards[cnt - 1 : : -1]:
            if v % 2 != x %2:
                ans = max(ans,replaced_sum(v))
                break
        return ans