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

Posted 2023-03-17 rose_halo

题目描述：

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

 

题解：

public int largestSumAfterKNegations(int[] nums, int k) 
        // 首先要为大的值取反，先对nums数组按照绝对值从大到小排序
        nums = IntStream.of(nums) //转换成IntStream流
                        .boxed()  //int -> Integer
                        .sorted((a, b) -> Math.abs(b) - Math.abs(a)) //排序
                        .mapToInt(Integer::intValue).toArray();
        int len  = nums.length;
        for (int i = 0; i < len; i++) 
            if (nums[i] < 0 && k > 0) 
                nums[i] = -nums[i];
                k--;
            
        
        // k的最大值和数组长度是一样长的，遍历完数组k还不为0，这时候就要判断，如果k是偶数，其实不用管，一个数来回反转偶数次，还是它自己。如果是奇数，就要管了，这是就可以把绝对值最小的那个数取反，代价最小。
        if (k % 2 == 1) nums[len-1] = - nums[len-1];
        return IntStream.of(nums).sum();