結果

問題 No.1470 Mex Sum
ユーザー da_ab
提出日時 2021-04-13 20:23:11
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 508 bytes
コンパイル時間 88 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 64,192 KB
最終ジャッジ日時 2024-06-30 01:00:40
合計ジャッジ時間 51,803 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 13 WA * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 1470 Mex Sum

from scipy.special import comb
from sys import stdin

n=int(stdin.readline())
a=tuple(map(int,stdin.readline().split()))

classifiedA=[0]*(3+1)
for e in a:
    if e==1:
        classifiedA[1]+=1
    elif e==2:
        classifiedA[2]+=1
    else:
        classifiedA[3]+=1
        
sumA=0
sumA+=3*classifiedA[1]*classifiedA[2]
sumA+=2*(comb(classifiedA[1],2,exact=True)+(classifiedA[1]*classifiedA[3]))
sumA+=1*(comb(classifiedA[2],2,exact=True)+(classifiedA[2]*classifiedA[3]))
print(sumA)
0