結果

問題 No.1708 Quality of Contest
ユーザー ygd.ygd.
提出日時 2021-10-19 23:53:44
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,174 ms / 2,000 ms
コード長 2,544 bytes
コンパイル時間 146 ms
コンパイル使用メモリ 81,724 KB
実行使用メモリ 149,240 KB
最終ジャッジ日時 2023-10-20 10:54:17
合計ジャッジ時間 19,570 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
53,404 KB
testcase_01 AC 40 ms
53,404 KB
testcase_02 AC 40 ms
53,404 KB
testcase_03 AC 145 ms
76,992 KB
testcase_04 AC 150 ms
77,292 KB
testcase_05 AC 157 ms
77,588 KB
testcase_06 AC 121 ms
77,052 KB
testcase_07 AC 137 ms
77,028 KB
testcase_08 AC 40 ms
53,412 KB
testcase_09 AC 465 ms
136,936 KB
testcase_10 AC 661 ms
143,244 KB
testcase_11 AC 1,050 ms
142,956 KB
testcase_12 AC 954 ms
146,844 KB
testcase_13 AC 827 ms
142,804 KB
testcase_14 AC 979 ms
146,600 KB
testcase_15 AC 855 ms
146,140 KB
testcase_16 AC 953 ms
149,240 KB
testcase_17 AC 957 ms
144,728 KB
testcase_18 AC 932 ms
144,752 KB
testcase_19 AC 1,006 ms
142,616 KB
testcase_20 AC 871 ms
144,032 KB
testcase_21 AC 1,174 ms
146,712 KB
testcase_22 AC 1,045 ms
143,240 KB
testcase_23 AC 1,031 ms
143,260 KB
testcase_24 AC 750 ms
143,236 KB
testcase_25 AC 744 ms
146,604 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
#input = sys.stdin.readline
input = sys.stdin.buffer.readline
from heapq import heapify, heappop, heappush

def main():
    N,M,X = map(int,input().split())
    A = []; B = []
    not_used = []; used = []
    for _ in range(N):
        a,b = map(int,input().split())
        A.append(a)
        B.append(b)
        heappush(not_used,(-a,b)) #値、組
    K = int(input())
    C = list(map(int,input().split()))
    dic = {}
    for c in C:
        if c in dic:
            dic[c] += 1
        else:
            dic[c] = 1
    L = []
    used_group = set([])
    score = 0
    ans = 0
    for i in range(N):
        if used:
            a,b = heappop(used)
            val_used_group = (-a,b)
        else:
            val_used_group = (0,0) #dummy
        
        val_not_group = (0,0) #dummy
        while not_used:
            a,b = heappop(not_used)
            a *= -1 #正に値に直す
            #print(a,b)
            #print("VAL",val_used_group)
            if b not in used_group:
                val_not_group = (a,b)
                break
            else:
                if val_used_group[0] > a: #既に出した値が同じ組中で最高
                    heappush(used,(-a,b))
                else:
                    temp = (a,b)
                    val_used_group, temp = temp, val_used_group
                    a,b = temp
                    heappush(used,(-a,b))
        #print(val_used_group,val_not_group)
        if val_used_group == (0,0):
            L.append(val_not_group)
            used_group.add(val_not_group[1])
            score += val_not_group[0] + X
        elif val_not_group == (0,0):
            L.append(val_used_group)
            used_group.add(val_used_group[1])
            score += val_used_group[0]
        else:
            score_used = score + val_used_group[0]
            socre_not = score + val_not_group[0] + X #1組増えるので
            if score_used > socre_not: #使用済みのグループを使う。
                score = score_used
                L.append(val_used_group)
                used_group.add(val_used_group[1])
                heappush(not_used,(-val_not_group[0],val_not_group[1]))
            else:
                score = socre_not
                L.append(val_not_group)
                used_group.add(val_not_group[1])
                heappush(used, (-val_used_group[0],val_used_group[1]))
        #print(score)
        if i+1 in dic:
            ans += score*dic[i+1]
    #print(L)
    print(ans)


if __name__ == '__main__':
    main()
0