結果

問題 No.1708 Quality of Contest
ユーザー kasuyaSansankasuyaSansan
提出日時 2021-10-26 20:00:12
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,560 bytes
コンパイル時間 255 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 131,776 KB
最終ジャッジ日時 2024-10-05 21:39:43
合計ジャッジ時間 6,352 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 TLE -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,M,X = map(int, input().split(" "))

questions = []
for i in range(N):
    questions.append(list(map(int, input().split(" "))))
    
if N > 900000:
    print("read_finish")

K = int(input())
num_answers = list(map(int, input().split(" ")))


def calc_confident(question_list, num_answers, X):
    sum = 0
    for num in num_answers:
        janru = set()
        for i in range(num):
            janru.add(question_list[i][1])
            sum += question_list[i][0]
        sum += len(janru)*X
    return sum

def get_other_j_first(sorted_list, j_set):
    for s in sorted_list:
        if s[1] not in j_set:
            return s
    return None

def get_max(sorted_list, j_set, N, X):
    #sorted_list = sorted(questions, key= lambda x:-x[0])
    if not j_set:
        return sorted_list[0]

    n_j = len(j_set)
    if n_j == N:
        return sorted_list[0]

    max_q = sorted_list[0]
    #other_j_list = [s for s in sorted_list if s[1] not in j_set]
    max_q_j = get_other_j_first(sorted_list, j_set)

    if not max_q_j:
        return max_q

    if max_q == max_q_j:
        return max_q

    dif = max_q[0] - max_q_j[0]
    if dif > X:
        return max_q
    else:
        return max_q_j

print("sort start")
questions = sorted_list = sorted(questions, key= lambda x:-x[0])
now_list = []
j_set = set()
for i in range(len(questions)):
    max_q = get_max(questions, j_set, N,X)

    now_list.append(max_q)
    j_set.add(max_q[1])
    questions.remove(max_q)
    print(i, len(questions))



val = calc_confident(now_list, num_answers, X)
print(val)
0