1005. K 次取反后最大化的数组和
LeetCode: 1005. K 次取反后最大化的数组和
简 单 \color{#00AF9B}{简单} 简单
给你一个整数数组
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
方法1:贪心 + 排序
对于数组中所有已存的负数,可以进行至多k
次的取反,每次取反要尽量选择更小的负数(贪心)。这样剩余的大的负数对最大和的负面影响更小;而值小的负数取反后变为值大的正数,对最大和的正面影响最大。
-
如果坐满了
k
次操作,不管数组中是否还有负数,此时就已经是可能的最大和了。因为剩余的负数是对最大和负面影响最小的了。 -
若将所有负数都取反后,即此时数组中皆为正数,却还有剩余次数要做,设已经完成
done
次取反,则剩余的次数为k-done
,此时一共分为2种情况:- 如果
k-done
为偶数。由于【可以对同一个数进行多次的取反】,我们可以将其中的偶数次全都抽象地附加于任意的同一数上。由于一个数取偶数次的反仍是它本身,这样一来数组仍为原数组。又因为此时数组中都是正数,所以此时是确定的最大和了。 - 如果
k-done
为奇数。同理,将其中的偶数次附加于同一数上。这样一来就剩下了1
次取反,我们当然希望是将最小的正数取反,这样对最大和的负面影响才最小。
- 如果
为了更好的定位元素,我们可以先排序,排序时按照每个元素的绝对值从小到大进行。
#include <vector>
#include <algorithm>
#include <numeric>
using namespace std;
class Solution
{
public:
int largestSumAfterKNegations(vector<int> &nums, int k)
{
const int n = nums.size();
sort(nums.begin(), nums.end(),