K次取反后最大化的数组和

leetcode 1005.K次取反后最大化的数组和

题目链接

给定一个整数数组 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]。

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

1 <= A.length <= 10000
1 <= K <= 10000
-100 <= A[i] <= 100

思路

将元素按绝对值排序，从大到小遍历，当前k不为0且当前数为负则取反，剩下的k > 0且为奇数就对最小的数取反 再累加

    static bool cmp(int a, int b)
    {
        return abs(a) < abs(b);
    }
    int largestSumAfterKNegations(vector<int>& A, int K) {
        sort(A.begin(), A.end(), cmp);
        int result = 0;
        for(int i = A.size() - 1; i >= 0 && K; i--)
        {
            if(A[i] < 0)
            {
                A[i] = -A[i];
                K--;
            }
        }
        //k为奇数,把绝对值最小的数取反
        if(K%2)
            A[0] = -A[0];
        for(auto it : A){
            result += it;
        }      
        return result;    
    }