結果
問題 |
No.1470 Mex Sum
|
ユーザー |
|
提出日時 | 2021-04-09 22:50:23 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 427 bytes |
コンパイル時間 | 302 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 96,896 KB |
最終ジャッジ日時 | 2024-06-25 06:31:50 |
合計ジャッジ時間 | 11,122 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | WA * 10 TLE * 1 -- * 38 |
ソースコード
N = int(input()) A = sorted([int(i) for i in input().split()]) 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 + 2*((N-1)-j+1) break else: sum = sum + int(((N-1)-i+1)*(N-1)/2) break print(sum)