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

小鱼在乎2024-08-27 10:21

题目：

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

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

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

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

思路：先将数组中所有负数按照绝对值由大到小依次取反，如果此时k还不为0，则反复将最小元素取反直至k为0。

代码：

java 复制代码

    public int largestSumAfterKNegations(int[] nums, int k) {
        //将数组按照绝对值大小由大到小排序
        nums=IntStream.of(nums)
                      .boxed()
                      .sorted((a,b)->Math.abs(b)-Math.abs(a))
                    .mapToInt(Integer::intValue).toArray();
        for(int i=0;i<nums.length;i++){
            if(nums[i]<0&&k>0){
                nums[i]=-nums[i];
                k--;
            }
        }
        //如果K不为0，则反复反转最小的元素
        if(k%2==1) nums[nums.length-1]=-nums[nums.length-1];
        return Arrays.stream(nums).sum();
    }

注释：

nums=IntStream.of(nums)

.boxed()

.sorted((a,b)->Math.abs(b)-Math.abs(a))

.mapToInt(Integer::intValue).toArray();

1.转换为流：IntStream.of(nums)将给定的nums数组转换为IntStream

2.装箱：boxed()将IntStream中的基本类型int转换为Stream<Integer>，即包装为Integer对象

3.排序：sorted((a,b)->Math.abs(b)-Math.abs(a))根据绝对值大小进行降序排序

4.拆箱：mapToInt(Integer::intValue)将Integer对象转换为IntStream

5.收集为数组：toArray()将IntStream中的元素收集到一个数组中

Arrays.stream(nums).sum()

1.创建流：通过Arrays.stream(nums)，将数组nums转换为一个流对象。

2.计算和：利用流的sum()方法，对流中的每个元素进行求和操作。