LeetCode每日一题 2600. K 件物品的最大和

2600. K 件物品的最大和

题意

袋子中装有一些物品，每个物品上都标记着数字 1 、0 或 -1 。

给你四个非负整数 numOnes 、numZeros 、numNegOnes 和 k 。

袋子最初包含：

numOnes 件标记为 1 的物品。
numZeroes 件标记为 0 的物品。
numNegOnes 件标记为 -1 的物品。
现计划从这些物品中恰好选出 k 件物品。返回所有可行方案中，物品上所标记数字之和的最大值。

样例输入

numOnes = 3, numZeros = 2, numNegOnes = 0, k = 2

样例输出

2

提示

0 <= numOnes, numZeros, numNegOnes <= 50
0 <= k <= numOnes + numZeros + numNegOnes

c++代码

class Solution {
public:
    int kItemsWithMaximumSum(int numOnes, int numZeros, int numNegOnes, int k) {
        if(numOnes >= k) return k;
        else if(numOnes + numZeros >= k) return numOnes;
        else return numOnes - (k - numOnes - numZeros);
    }
};

python代码

class Solution:
    def kItemsWithMaximumSum(self, numOnes: int, numZeros: int, numNegOnes: int, k: int) -> int:
        if numOnes > k:
            return k
        elif numOnes + numZeros > k:
            return numOnes
        else:
            return numOnes - (k - numOnes - numZeros)

备注

简单题