#708252 (PyPy3) No.1708 Quality of Contest

• 問題: No.1708 Quality of Contest
• ユーザー: ygd.
• 提出日時: 2021-10-19 23:53:44
• 言語: PyPy3 (7.3.15)
• 結果: AC
• 実行時間: 1,035 ms / 2,000 ms
• コード長: 2,544 bytes
• コンパイル時間: 277 ms
• コンパイル使用メモリ: 82,100 KB
• 実行使用メモリ: 149,620 KB
• 最終ジャッジ日時: 2024-09-20 06:28:44
• 合計ジャッジ時間: 17,520 ms
• ジャッジサーバーID （参考情報）: judge1 / judge3

ファイルパターン	結果
sample	AC * 3
other	AC * 23

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
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()
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX