結果
問題 |
No.1470 Mex Sum
|
ユーザー |
|
提出日時 | 2021-12-13 01:13:25 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 245 bytes |
コンパイル時間 | 191 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 100,608 KB |
最終ジャッジ日時 | 2024-07-21 14:49:01 |
合計ジャッジ時間 | 7,119 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 14 WA * 35 |
ソースコード
n = int(input()) A = list(map(int, input().split())) one = 0 two = 0 for a in A: if a == 1: one += 1 elif a == 2: two += 1 x = n - one ans = x * (x - 1) // 2 ans += 2 * one * (one - 1) // 2 ans += 3 * one * two print(ans)