LeetCode 1005[贪心] K次取反后最大化的数组和 HERODING的LeetCode之路

Posted 2021-12-28 HERODING23

解题思路

一道很简单的贪心题目，分析一波就能有很好的思路。首先为了让翻转K次后和最大，那么最好是能把所有的负数都转变成正数，不能全部转变就直接返回和，能够全部转变并且K还有剩余，讨论K的奇偶性，偶数直接返回之前的求和即可（翻转抵消），奇数的话那肯定会有负数出现，那么只要把最小的值翻转成负数即可，代码的注释已经很详尽了，代码如下：

代码

class Solution 
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) 
        int total = accumulate(nums.begin(), nums.end(), 0);
        sort(nums.begin(), nums.end());
        // 尽可能把所有负数变成正数，如果还有负数，说明K已经用完，没有负数，K剩余
        for(int i = 0; i < nums.size(); i ++) 
            if(k == 0 || nums[i] >= 0) 
                break;
            
            if(nums[i] < 0) 
                k --;
                nums[i] *= -1;
                total += 2 * nums[i];
            
        
        // K如果剩余且是奇数，此时数组都是正数，那么只要把最小的值翻转成负数即可
        int minNum = *min_element(nums.begin(), nums.end());
        return k % 2 == 0 ? total : (total - 2 * minNum);
    
;