【力扣】LCP40.心算挑战

题目描述

「力扣挑战赛」心算项目的挑战比赛中，要求选手从 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

解释：不存在获取有效得分的卡牌方案。

提示：

• 1 <= cnt <= cards.length <= 10^5
• 1 <= cards[i] <= 1000

AC代码

class Solution {
public:
    int maximumScore(vector<int>& cards, int cnt) {
        if (cards.size() == 1) {
            if (cards[0] % 2) return 0;
            else return cards[0];
        }
        int sum = 0, odd = 0, even = 0, re = 0;
        if (cards.size() == cnt) {
            for (int i = 0; i < cards.size(); i++) {
                sum += cards[i];
            }
            if (sum % 2) return 0;
            else return sum;
        }
        //vector中的sort方法排序
        sort(cards.begin(), cards.end()); //从大到小排
        for (int i = cards.size() - 1; i >= cards.size() - cnt; i--) { //后cnt个数之和
            sum += cards[i];
            if (cards[i] % 2) odd = cards[i];
            else even = cards[i];
        }
        if (!(sum % 2)) return sum; //偶数直接返回
        else {
            for (int i = cards.size()-cnt-1; i>=0; i--) {
                if (!(cards[i] % 2) && odd)
                    re = max(re, sum - odd + cards[i]);
                if ((cards[i] % 2) && even)
                    re = max(re, sum - even + cards[i]); 
            }
        }
        return re;
    }
};