【贪心】K 次取反后最大化的数组和

【贪心】K 次取反后最大化的数组和

题目链接

题目描述：
给你一个整数数组 nums 和一个整数 k ，按以下方法修改该数组：
选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。
以这种方式修改数组后，返回数组 可能的最大和 。

提示
1 <= nums.length <= 104
-100 <= nums[i] <= 100
1 <= k <= 104

算法思路

三种情况讨论：
先排序；负数个数m
if m>k  反转前k个数，再相加。
if m==k 把所有的数都当成正数加起来。
if m<k
	if (m-k)%2==0 把所有的数都当成正数加起来。
	else 把第一个为负后，之后所有的数都当成正数加起来。
在这里插入代码片

代码如下

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        int l = nums.size(), cnt = 0, sum = 0;
        for (int i = 0; i < l; i++) {
            if (nums[i] < 0) cnt++;
        }
        sort(nums.begin(), nums.end());
        if (cnt == k || (cnt < k && (k - cnt) % 2 == 0)) {
            for (int i = 0; i < l; i++){
                sum += abs(nums[i]);
            }

        }
        else if(cnt>k) {
            for (int i = 0; i < k; i++)
            sum+=abs(nums[i]);
            for (int j = k; j <l ; j++)
            sum += nums[j];
        }
        else if((k-cnt)%2!=0){
            for(int i=0;i<l;i++)  nums[i]=abs(nums[i]);
            sort(nums.begin(), nums.end());
            sum-=nums[0];
            for(int i=1;i<l;i++) sum+=nums[i];
        }
        return sum;
    }
};