1005.K次取反后最大化的数组和-贪心算法

力扣题目链接(opens new window)

题目描述

给定一个整数数组 A，我们只能用以下方法修改该数组：我们选择某个索引 i 并将 A[i] 替换为 -A[i]，然后总共重复这个过程 K 次。（我们可以多次选择同一个索引 i。）

以这种方式修改数组后，返回数组可能的最大和。

示例 1：

• 输入：A = [4,2,3], K = 1
• 输出：5
• 解释：选择索引 (1) ，然后 A 变为 [4,-2,3]。

示例 2：

• 输入：A = [3,-1,0,2], K = 3
• 输出：6
• 解释：选择索引 (1, 2, 2) ，然后 A 变为 [3,1,0,2]。

思路

贪心的思路，局部最优：让绝对值大的负数变为正数，当前数值达到最大，整体最优：整个数组和达到最大。

C++代码

class Solution {
static bool cmp(int x, int y){
        return abs(x) > abs(y);
    }
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end(), cmp);
        for(int i = 0; i < nums.size(); i++){
            if(nums[i] < 0 && k > 0){
                nums[i] = -nums[i];
                k--;
            }
        }
        if(k%2 == 1) nums[nums.size()-1] *= -1;
        int result = 0;
        for(int i = 0; i < nums.size(); i++){
            result += nums[i];
        }
        return result;
    }
};

编程问题：

• 如何按绝对值排序

之前只会用sort从小到大排序，第一次发现可以自定义排序准则，这个知识点可以参考C++ sort()排序详解-CSDN博客，讲得很清楚。

• 排序准则函数的定义

平时用static比较少，第一次写cmp函数没有加static修饰，结果就报错了。然后尝试把他放到public里面，还是报错。只有加static才行。看了static的详解还是没能理解这里为什么必须用static。C/C++ static关键字详解（最全解析，static是什么，static如何使用，static的常考面试题）-CSDN博客