結果
| 問題 | No.1470 Mex Sum | 
| コンテスト | |
| ユーザー |  | 
| 提出日時 | 2021-12-06 10:52:31 | 
| 言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) | 
| 結果 | 
                                WA
                                 
                             | 
| 実行時間 | - | 
| コード長 | 314 bytes | 
| コンパイル時間 | 95 ms | 
| コンパイル使用メモリ | 12,544 KB | 
| 実行使用メモリ | 19,004 KB | 
| 最終ジャッジ日時 | 2024-07-07 08:51:56 | 
| 合計ジャッジ時間 | 7,531 ms | 
| ジャッジサーバーID (参考情報) | judge2 / judge1 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 1 WA * 1 | 
| other | AC * 1 WA * 48 | 
ソースコード
N = int(input())
A = list(map(int,input().split()))
count1 = 0
count2 = 0
count = 0
for i in A:
    if i == 1:count1 += 1
    elif i == 2:count2 += 2
    else:count += 1
S = count1 * (count1 - 1) + 3 * count1 * count2 + 2 * count1 * count + count2 * (count2 - 1) // 2 + count2 * count + count * (count-1)
print(S)
            
            
            
        