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

Posted 2021-05-24 jtwty

目录

• • 一.问题
  • 二.分析
  • 三.代码

一.问题

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

来源：力扣（LeetCode）
链接：题目描述

二.分析

（1）首先整理题意：我们要经过k次取反操作然后使得数组和最大。

（2）要想使数组和最大，那么我们首先可以将负数变为正数，所以我们先对数组从小到大进行排序。然后使用一个for循环，直到k次机会用完或者遍历一遍完成后退出循环。

（3）接下来如果k次机会已经用完或者剩下机会是偶数个，那么我们直接求和返回。反之，我们再对数组进行一次排序（这时数组里面的元素均为非负数），这个时候因为都是非负数，所以我们只要对最小的那个数操作就好了，因为上面已经排除了剩下偶数个机会的情况，所以此时将最小数直接取反求和返回即可。

三.代码

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        //先从小到大排一次序
        sort(nums.begin(),nums.end());
        //在k满足的情况下将负数变为正数
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]<0 && k>0)
            {
                k--;
                nums[i]=-nums[i];
            }
        }
        if(k%2!=0) {sort(nums.begin(),nums.end());nums[0]=-nums[0];}
        //求和
        int sum=0;
        for(int i=0;i<nums.size();i++)
        {
            sum+=nums[i];
        }
        return sum;
    }
};