結果
問題 | No.120 傾向と対策:門松列(その1) |
ユーザー | 1235486527 |
提出日時 | 2019-12-22 16:02:57 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 104 ms / 5,000 ms |
コード長 | 536 bytes |
コンパイル時間 | 160 ms |
コンパイル使用メモリ | 10,952 KB |
実行使用メモリ | 8,320 KB |
最終ジャッジ日時 | 2023-10-12 09:27:00 |
合計ジャッジ時間 | 1,434 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 98 ms
8,244 KB |
testcase_01 | AC | 104 ms
8,316 KB |
testcase_02 | AC | 63 ms
8,320 KB |
testcase_03 | AC | 104 ms
8,216 KB |
ソースコード
import heapq def f(): _ = input() D = {} for L in map(int, input().split()): D[L] = D.setdefault(L, 0) + 1 hq = [-v for v in D.values()] heapq.heapify(hq) kl = [None] * 3 ans = 0 while True: if len(hq) < 3: print(ans) return for i in range(3): kl[i] = heapq.heappop(hq) + 1 for i in range(3): if kl[i] != 0: heapq.heappush(hq, kl[i]) ans += 1 return T = int(input()) for _ in range(T): f()