【leetcode刷题】T181-K 次取反后最大化的数组和

示例 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]。

class Solution(object):
    def largestSumAfterKNegations(self, A, K):
        """
        :type A: List[int]
        :type K: int
        :rtype: int
        """
        A.sort()
        count_neg = len(filter(lambda x: x<0, A))
        # 更改前min(K, count_neg)个负数为相反数
        if count_neg >= K:
            A[:K] = list(map(lambda x: -x, A[:K]))
        else:
            A[:count_neg] = list(map(lambda x: -x, A[:count_neg]))
            # 全是正数，重新排序
            A.sort()
            if (K - count_neg) % 2 == 1:
                A[0] = -A[0]
        return sum(A)

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& A, int K) {
        sort(A.begin(), A.end());
        // 计算负数个数
        int count_neg = 0;
        for(int i=0; i<A.size(); i++){
            if(A[i] >= 0)
                break;
            count_neg++;
        }
        // 更改前min(K, count_neg)个负数为相反数
        if(count_neg >= K){
            for(int i=0; i<K; i++)
                A[i] = -A[i];
        }else{
            for(int i=0; i<count_neg; i++)
                A[i] = -A[i];
            // 重新排序
            sort(A.begin(), A.end());
            if((K - count_neg) % 2 == 1)
                A[0] = -A[0];
        }
        // 求和
        int res = 0;
        for(int i=0; i<A.size(); i++)
            res += A[i];
        return res;
    }
};