class Solution {
public:
int kItemsWithMaximumSum(int numOnes, int numZeros, int numNegOnes, int k) {
}
};
2600. K 件物品的最大和
袋子中装有一些物品，每个物品上都标记着数字 1 、0 或 -1 。
给你四个非负整数 numOnes 、numZeros 、numNegOnes 和 k 。
袋子最初包含：
numOnes 件标记为 1 的物品。numZeroes 件标记为 0 的物品。numNegOnes 件标记为 -1 的物品。现计划从这些物品中恰好选出 k 件物品。返回所有可行方案中，物品上所标记数字之和的最大值。
示例 1：
输入：numOnes = 3, numZeros = 2, numNegOnes = 0, k = 2
输出：2
解释：袋子中的物品分别标记为 {1, 1, 1, 0, 0} 。取 2 件标记为 1 的物品，得到的数字之和为 2 。
可以证明 2 是所有可行方案中的最大值。
示例 2：
输入：numOnes = 3, numZeros = 2, numNegOnes = 0, k = 4
输出：3
解释：袋子中的物品分别标记为 {1, 1, 1, 0, 0} 。取 3 件标记为 1 的物品，1 件标记为 0 的物品，得到的数字之和为 3 。
可以证明 3 是所有可行方案中的最大值。
提示：
0 <= numOnes, numZeros, numNegOnes <= 500 <= k <= numOnes + numZeros + numNegOnes原站题解
golang 解法, 执行用时: 0 ms, 内存消耗: 2.6 MB, 提交时间: 2023-03-29 14:50:36
func kItemsWithMaximumSum(numOnes, numZeros, _, k int) int {
if k <= numOnes {
return k
}
if k <= numOnes+numZeros {
return numOnes
}
return numOnes*2 + numZeros - k
}
python3 解法, 执行用时: 52 ms, 内存消耗: 15 MB, 提交时间: 2023-03-29 14:49:53
class Solution:
def kItemsWithMaximumSum(self, numOnes: int, numZeros: int, numNegOnes: int, k: int) -> int:
if k <= numOnes: return k
elif k <= numOnes + numZeros: return numOnes
else: return numOnes * 2 + numZeros - k
python3 解法, 执行用时: 32 ms, 内存消耗: 14.9 MB, 提交时间: 2023-03-29 14:42:23
class Solution:
def kItemsWithMaximumSum(self, numOnes: int, numZeros: int, numNegOnes: int, k: int) -> int:
nums = []
for i in range(numOnes):
nums.append(1)
for i in range(numZeros):
nums.append(0)
for i in range(numNegOnes):
nums.append(-1)
nums.sort(reverse=True)
res = 0
for i in range(k):
res += nums[i]
return res