結果
問題 |
No.1470 Mex Sum
|
ユーザー |
![]() |
提出日時 | 2021-04-09 21:27:13 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 87 ms / 2,000 ms |
コード長 | 474 bytes |
コンパイル時間 | 286 ms |
コンパイル使用メモリ | 82,368 KB |
実行使用メモリ | 103,904 KB |
最終ジャッジ日時 | 2024-06-25 04:07:42 |
合計ジャッジ時間 | 5,583 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 49 |
ソースコード
mod = 1000000007 eps = 10**-9 def main(): import sys input = sys.stdin.readline def comb(a): return a * (a-1) // 2 N = int(input()) A = list(map(int, input().split())) x = 0 y = 0 for a in A: if a == 1: x += 1 elif a == 2: y += 1 z = N - x - y ans = 2 * comb(x) + 3 * x * y + 2 * x * z + 1 * comb(y) + 1 * y * z + 1 * comb(z) print(ans) if __name__ == '__main__': main()