题目描述
给定一个整数数组 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
解题思路
第一次想到的解决办法是先对原始数组排序,然后从最小的开始判断,分别对正数、负数、零做不同的处理。但是再编写过程中突然想到K不一定比A.size()小,而且一个索引可以多次取反,所以每次取反后仍然要保持数组的有序性,所以改成使用优先级队列。
int largestSumAfterKNegations(vector<int>& A, int K) {
priority_queue<int,vector<int>,greater<int>> que;
int N = A.size();
int sum = 0;
for(int i=0;i<N;i++) que.push(A[i]);
for(int i=1;i<=K;i++){
int tmp = que.top();
que.pop();
que.push(-tmp);
}
while(!que.empty()){
sum+=que.top();
que.pop();
}
return sum;
}