結果

問題 No.1470 Mex Sum
ユーザー da_ab
提出日時 2021-04-13 20:36:01
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,135 ms / 2,000 ms
コード長 542 bytes
コンパイル時間 300 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 63,816 KB
最終ジャッジ日時 2024-06-30 01:08:11
合計ジャッジ時間 58,329 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
外部呼び出し有り
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

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)+comb(classifiedA[3],2,exact=True)+(classifiedA[2]*classifiedA[3]))
print(sumA)
0