結果
問題 |
No.1470 Mex Sum
|
ユーザー |
|
提出日時 | 2022-01-19 00:23:41 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 335 bytes |
コンパイル時間 | 475 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 19,276 KB |
最終ジャッジ日時 | 2024-11-23 13:38:41 |
合計ジャッジ時間 | 8,145 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 14 WA * 35 |
ソースコード
n = int(input()) A = list(map(int, input().split())) L = [0, 0, 0, 0] for i in range(n): if A[i] <= 2: L[A[i]] += 1 else: L[3] += 1 ans = 0 ans += L[1] * (L[1] - 1) ans += L[2] * (L[2] - 1) // 2 + L[3] * (L[3] - 1) // 2 ans += 3 * L[1] * L[2] ans += L[1] * L[3] ans += L[2] * L[3] print(ans)