LeetCode LCP 40. 心算挑战（贪心、排序）

最新推荐文章于 2024-09-06 09:22:51 发布

岁忧

最新推荐文章于 2024-09-06 09:22:51 发布

阅读量351

点赞数 9

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展 c++ 贪心算法数据结构

本文链接：https://blog.csdn.net/weixin_46028214/article/details/140835632

版权

LeetCode 专栏收录该内容

436 篇文章 0 订阅

订阅专栏

题目：LCP 40. 心算挑战

在这里插入图片描述

思路：先将数组cards按降序排序，然后获得前cnt个数的累积和sum。
当sum为偶数直接输出；当sum为奇数时，就需要找到前cnt个元素中的最小的奇数p1和偶数p0，同时找到剩余数中最大的奇数q1和偶数q0。
然后进行两种交换：sum-p0+q1和sum-p1+q0

class Solution {
public:
    int maxmiumScore(vector<int>& cards, int cnt) {
        sort(cards.begin(),cards.end(),greater<>());
        int sum=0;
        int p0=-1,p1=-1;
        for(int i=0;i<cnt;i++){
            sum+=cards[i];
            if(cards[i]%2) p1=cards[i];
            else p0=cards[i];
        }
        if(sum%2==0) return sum;
        int q0=-1,q1=-1;
        for(int i=cnt;i<cards.size();i++){
            if(cards[i]%2&&q1==-1) q1=cards[i];
            else if(cards[i]%2==0&&q0==-1) q0=cards[i];
        }
        bool flag=0;
        int t1=0,t2=0;
        if(p0!=-1&&q1!=-1){
            t1=sum-p0+q1;
            flag=1;
        }
        if(p1!=-1&&q0!=-1){
            t2=sum-p1+q0;
            flag=1;
        }
        if(flag) return max(t1,t2);
        else return 0;
    }
};