結果

問題 No.1470 Mex Sum
コンテスト
ユーザー DrDrpilot
提出日時 2022-02-23 13:50:43
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 116 ms / 2,000 ms
コード長 293 bytes
コンパイル時間 254 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 19,128 KB
最終ジャッジ日時 2024-07-01 13:37:10
合計ジャッジ時間 7,594 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

n=int(input())
a=list(map(int,input().split()))
num_1=num_2=num_3=0
for i in a:
    if i==1:
        num_1+=1
    elif i==2:
        num_2+=1
    else:
        num_3+=1
ans=num_1*(num_1-1)*2//2+num_2*(num_2-1)*1//2+num_3*(num_3-1)*1//2
ans+=num_1*num_2*3+num_2*num_3*1+num_3*num_1*2
print(ans)
0