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 <= 104
  • -100 <= nums[i] <= 100
  • 1 <= k <= 104

思路

先按从小到大顺序排序,遍历数组,遇到负数就取反,同时记录翻转次数,如果遇到非负数则跳出循环,或者反转次数到达k了,也跳出循环。

这时记录一下剩余反转次数,如果是偶数,直接把当前数组加和,如果是奇数,再从小到达排列数组,把第一个数取反,再把数组加和。

对应C++代码如下:

class Solution {
public:
    int getSum(vector<int> num)
    {
        int sum=0;
        for(int i=0;i<num.size();i++)
        {
            sum+=num[i];
        }
        return sum;
    }
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        sort(nums.begin(),nums.end());
        int count=0;
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]<0)
            {
                if(count==k)
                {
                    break;
                }
                else
                {
                    count++;
                    nums[i]=-nums[i];
                }
            }
            else{
                break;
            }
        }
        int resK=k-count;
        if(resK%2==0)
        {
            return getSum(nums);
        }
        else{
            sort(nums.begin(),nums.end());
            nums[0]=-nums[0];
            return getSum(nums);
        }       
    }
};

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值