結果
問題 | No.120 傾向と対策:門松列(その1) |
ユーザー | H3PO4 |
提出日時 | 2020-04-29 19:06:53 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 135 ms / 5,000 ms |
コード長 | 543 bytes |
コンパイル時間 | 117 ms |
コンパイル使用メモリ | 12,416 KB |
実行使用メモリ | 10,880 KB |
最終ジャッジ日時 | 2024-05-07 01:51:15 |
合計ジャッジ時間 | 1,260 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 126 ms
10,752 KB |
testcase_01 | AC | 129 ms
10,880 KB |
testcase_02 | AC | 81 ms
10,752 KB |
testcase_03 | AC | 135 ms
10,880 KB |
ソースコード
from collections import Counter import heapq T = int(input()) for _ in range(T): N = int(input()) L = Counter(map(int, input().split())).values() V = [-x for x in L] heapq.heapify(V) ans = 0 tmp_lst = [0] * 3 while True: try: for i in range(3): tmp_lst[i] = heapq.heappop(V) ans += 1 except IndexError: print(ans) break for i in range(3): if tmp_lst[i] < -1: heapq.heappush(V, tmp_lst[i] + 1)