結果
問題 | No.1470 Mex Sum |
ユーザー |
|
提出日時 | 2021-04-10 13:11:04 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 491 bytes |
コンパイル時間 | 221 ms |
コンパイル使用メモリ | 82,144 KB |
実行使用メモリ | 96,300 KB |
最終ジャッジ日時 | 2024-06-26 02:22:19 |
合計ジャッジ時間 | 7,856 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | RE * 2 |
other | RE * 49 |
ソースコード
from sys import stdin N = stdin.readline().rstrip() A = stdin.readline().rstrip().split() A = sorted([int(i) for i in A]) sum = 0 for i in range(N): if A[i] == 1: for j in range(i+1,N): if A[j] == 1: sum += 2 elif A[j] == 2: sum += 3 else: sum = sum + 2*((N-1)-j+1) break elif A[i] == 2: for j in range(i+1,N): if A[j] == 2: sum += 1 else: sum = sum + 1*((N-1)-j+1) break else: sum = sum + int(((N-1)-i+1)*((N-1)-i)/2) break print(sum)