結果

問題 No.1708 Quality of Contest
コンテスト
ユーザー wgrape
提出日時 2024-11-07 16:46:44
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,825 bytes
コンパイル時間 169 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 179,876 KB
最終ジャッジ日時 2024-11-07 16:46:59
合計ジャッジ時間 12,489 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 5 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

from collections import defaultdict
dic = defaultdict(list) # 種類:値

for _ in range(N):
    a,b = map(int,input().split())
    dic[b].append(a)

# 各種類の最大のものを取り出し[値, 種類]で管理
# 値の順に取り出せるようにする
# 以降は「新しい種類を取り出す」と「既存の種類の最大値」の比較
tops = []
for key, value in dic.items():
    sval = sorted(value)
    tops.append([sval.pop(), key])
    if sval:
        dic[key] = sval

tops = sorted(tops, key = lambda x:x[0])

# 一方で、K人の管理

K = input()
C = list(map(int,input().split()))

S = [0] * (N + 1) # i問目まで解く人が何人いるか
for c in C:
    S[c] += 1
    
for i in range(N - 1, -1, -1):
    S[i] += S[i + 1]
    
ans = 0
import heapq as hq
q = []
# topsの末尾の問題を解いて、その値 + Xを得るか
# qの問題を解くか
for i in range(1, N + 1): # i問目を解く
    # S[i]人が解く
    if S[i] == 0:
        break
    base = 0
    if not tops: # topsが既に空の場合、qの先頭を解く
        val = hq.heappop(q)
        base -= val # 正負逆転
    elif not q: # qが空の場合、topsの先頭を解く
        p, kind = tops.pop()
        base += p + X # 種類のポイントを得る
        # 種類kindの値をすべてqに追加
        for point in dic[kind]:
            hq.heappush(q, -point) # 正負反転
    else: # topsもqも空ではない
        tops_point = tops[-1][0] + X
        q_point = q[0] * (-1)
        if tops_point > q_point:
            p, kind = tops.pop()
            base += p + X
            for point in dic[kind]:
                hq.heappush(q, -point)
        else:
            val = hq.heappop(q)
            base -= val
    ans += base * S[i]
print(ans)
0