結果

問題 No.1470 Mex Sum
ユーザー ygd.
提出日時 2021-04-09 21:47:34
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 115 ms / 2,000 ms
コード長 319 bytes
コンパイル時間 381 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 106,880 KB
最終ジャッジ日時 2024-06-25 04:55:56
合計ジャッジ時間 7,017 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict

N = int(input())
A = list(map(int,input().split()))
dic = defaultdict(int)
for x in A:
    dic[x] += 1

ans = N*(N-1)//2 #すべて1

ones = dic[1]*(dic[1]-1)//2
ans += ones

onetwo = dic[1]*dic[2]
ans += onetwo * 2

oneelse = dic[1]*(N - dic[1] - dic[2])
ans += oneelse

print(ans)
0