字节面试题——1005. K 次取反后最大化的数组和

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

给你一个整数数组 nums 和一个整数 k ，按以下方法修改该数组：

• 选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。

重复这个过程恰好 k 次。可以多次选择同一个下标 i 。

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

示例 1：

输入：nums = [4,2,3], k = 1
输出：5
解释：选择下标 1 ，nums 变为 [4,-2,3] 。

示例 2：

输入：nums = [3,-1,0,2], k = 3
输出：6
解释：选择下标 (1, 2, 2) ，nums 变为 [3,1,0,2] 。

示例 3：

输入：nums = [2,-3,-1,5,-4], k = 2
输出：13
解释：选择下标 (1, 4) ，nums 变为 [2,3,-1,5,4] 。

提示：

• 1 <= nums.length <= 10^4
• -100 <= nums[i] <= 100
• 1 <= k <= 10^4

贪心+哈希表

class Solution {
    public int largestSumAfterKNegations(int[] nums, int k) {
        // 创建一个哈希表，用于记录每个数字出现的频率
        Map<Integer, Integer> freq = new HashMap<>();
        // 遍历数组 nums，统计每个数字出现的频率
        for (int num : nums) {
            freq.put(num, freq.getOrDefault(num, 0) + 1);
        }
        
        // 计算数组 nums 的初始总和
        int ans = Arrays.stream(nums).sum();
        
        // 尝试从 -100 到 -1 的数字进行翻转
        for (int i = -100; i < 0; ++i) {
            // 如果哈希表中包含当前数字 i
            if (freq.containsKey(i)) {
                // 计算能够翻转的次数 ops，即取 k 和当前数字频率的较小值
                int ops = Math.min(k, freq.get(i));
                
                // 更新总和 ans，翻转 ops 次 i，相当于增加 (-i) * ops * 2
                ans += (-i) * ops * 2;
                
                // 更新哈希表中当前数字 i 的频率
                freq.put(i, freq.get(i) - ops);
                // 更新哈希表中翻转后数字 -i 的频率
                freq.put(-i, freq.getOrDefault(-i, 0) + ops);
                
                // 减少 k 的值
                k -= ops;
                // 如果 k 为 0，结束循环
                if (k == 0) {
                    break;
                }
            }
        }
        
        // 如果 k 大于 0 （即k大于100，负数不够用，需要反转正数）并且 k 是奇数且没有 0 存在于哈希表中
        if (k > 0 && k % 2 == 1 && !freq.containsKey(0)) {
            // 从 1 到 100 的数字中找到最小的正数并翻转一次
            for (int i = 1; i <= 100; ++i) {
                if (freq.containsKey(i)) {
                    ans -= i * 2;
                    break;
                }
            }
        }
        
        // 返回最终的总和
        return ans;
    }
}

• 时间复杂度： $O(n+C)$ ，其中 $n$ 是数组 nums 的长度， $C$ 是数组 nums 中元素的范围，本题中 $C=201$。我们需要 $O(n)$ 的时间使用桶或哈希表统计每个元素出现的次数，随后需要 $O(C)$ 的时间对元素进行操作。
• 空间复杂度： $O(C)$ ，即为桶或哈希表需要使用的空间。