小根堆+贪心 -leetcode-1005. K 次取反后最大化的数组和

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

题目描述

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

小根堆+贪心

思路：

• 每次都是从 nums 中取 min 值，并反转
• 将反转后的 min 再次插入 小根堆中
• 对 小根堆 中所有元素 sum

class Solution {
    public int largestSumAfterKNegations(int[] nums, int k) {
        PriorityQueue<Integer> queue = new PriorityQueue<>((o1, o2) -> o1 - o2); // 小根堆
        for (int i : nums) {
            queue.add(i);
        }
        System.out.println(queue);
        // 贪心：每次从新数组中取min反转，反转k次即可
        for (int i = 0; i < k; i++) {
            int min = queue.poll();
            min = -min; // 反转
            queue.add(min); // 反转后再次放回
        }
        System.out.println(queue);
        // 求总和
        int sum = 0;
        for (int i : queue) sum += i;
        return sum;
    }
}