幸运平衡问题解决方案产生错误的结果

Question

问题

Lena正在准备一场重要的编码竞赛，在此之前有许多连续的预赛。最初，她的运气平衡是0。她相信“节省运气”，并想检查她的理论。每场比赛都由两个整数描述，L[i]和T[i]：

• L[i]是与比赛相关的运气量。如果莉娜赢得比赛，她的运气余额将减少L[i];如果她失去了它，她的运气余额将增加L[i]。
• T[i] 表示比赛的重要性评级。如果比赛很重要，则等于 1，如果比赛不重要，则等于 0。

如果Lena在不超过< code>k场重要比赛中失利，她在所有预赛中获胜后的最大运气是多少？该值可能为负值。

请在此处找到完整描述

我的努力

这是我为这个问题写的代码。我添加了所有不重要的问题，并将所有重要问题的运气值存储在一个数组中。对数组进行排序。然后从她被允许获胜的问题中提取所有最小运气余额，减去这个余额，并从她输掉的比赛的其余部分中添加运气。

法典

#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the luckBalance function below.
nk = input().split()
n = int(nk[0])
k = int(nk[1])

contests = []

maximumLuckBalance = 0

for i in range(n):
    luckcontestRating = input().split()
    luck =  int(luckcontestRating[0])
    contestRating = int(luckcontestRating[1])
    #print("Contest[",i+1, "]:" + "luck: ", luck, "Contest Imp: ", contestRating)
    if contestRating == 0:
        maximumLuckBalance += luck
    else:
        contests.append(luck)
contests.sort()
#print(contests)
winAllowed = len(contests) - k
#print(winAllowed)
winningLucksum = sum(contests[:winAllowed])
#print(winningLucksum)
loosingLucksum = sum(contests[winAllowed:])
maximumLuckBalance = maximumLuckBalance - winningLucksum + loosingLucksum
print(maximumLuckBalance)

问题

然而，这段代码在一个测试用例中失败了(请看下面的输入)。我的解决方案有什么问题？

输入

97 58
105 0
103 0
106 1
106 1
103 0
103 1
105 1
106 1
105 0
104 0
103 0
102 0
104 0
105 0
104 0
102 1
104 0
106 1
104 1
101 1
105 0
103 0
104 0
106 0
102 1
103 0
102 0
103 1
106 0
104 1
101 1
101 1
106 0
103 1
103 0
104 1
101 0
105 1
105 0
104 1
105 0
106 0
104 0
105 0
101 1
106 1
105 0
103 0
104 1
101 1
106 1
104 0
106 1
105 0
103 1
101 0
103 0
101 0
105 1
104 1
104 1
105 1
105 1
103 0
101 0
104 1
106 1
105 1
105 0
106 1
104 1
105 1
103 1
102 1
106 0
101 0
105 1
104 1
103 1
106 1
101 0
106 1
103 0
106 1
102 1
103 0
101 1
102 1
101 1
104 0
106 0
102 0
104 0
105 0
105 0
102 1
103 1

预期产量

10069

Answer 1

你的代码的问题在于你假设k总是小于或等于评级竞赛的数量 - len（contests）。

例如，让我们考虑以下输入：

4 3

1 0 

2 0

3 1

4 1

正确的输出必须是:

10

但是您的代码会产生:

4

因为，winningLucksum=sum（竞赛[：-1]）=3和lo看作Lucksum=和（竞赛[-1：]）=4，所以最大LuckBalance=3-3 4=4。

因此，确保win的永远不会通过替换得到否定

winAllowed = len(contests) - k

用一行：

winAllowed = max(len(contests) - k, 0)

然后，您的解决方案将通过所有测试。