这个问题总的来说还是一个排序加判断的问题。
时间复杂度O(nlgn)空间复杂度O(1)
先把A从小到大排序,用的是STL自带的排序函数sort()。
从负数最小的数开始一个个翻成正数。如果K已经用完了就返回A元素的和。
如果K没有用完,那么此时的A的元素已经全部都是大于等于零的数了。
如果K此时是偶数,那么直接返回A元素的和,不需要排序,因为一个数翻转两次得到的还是原来的数。
如果K此时不是偶数,就把新的A再排序一遍。
此时如果第一个数是0那么直接返回A元素的和,因为0翻转奇数次得到的还是原来的数。
如果第一个数不是0,那么就把第一个数翻转,然后返回A元素的和,因为第一个数最小。
tips:
不知道为什么for循环的终止判断语句用&&把语句A和语句B连在一起就会有buffer overflow,总之在循环体内写成if语句就过了。
accumulate(A.begin(),A.end(),0);这个函数用来把vector里面的元素相加,第三个参数是初始值,此处设为0。
class Solution {
public:
int largestSumAfterKNegations(vector<int>& A, int K) {
if(A.empty()) return 0;
if(K == 0)return accumulate(A.begin() , A.end() , 0);
sort(A.begin(), A.end());
for(int i = 0; A[i] < 0; i++)
{
A[i] = -A[i];
K--;
if(K == 0) return accumulate(A.begin() , A.end() , 0);
if(i == A.size()-1) break;
}
if(K%2 == 0) return accumulate(A.begin() , A.end() , 0);
sort(A.begin(), A.end());
if( A[0] == 0) return accumulate(A.begin() , A.end() , 0);
A[0] = -A[0];
return accumulate(A.begin() , A.end() , 0);
}
};