結果

問題 No.1470 Mex Sum
ユーザー AAAR Salmon
提出日時 2021-05-15 17:05:55
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 101 ms / 2,000 ms
コード長 522 bytes
コンパイル時間 183 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 19,200 KB
最終ジャッジ日時 2024-10-03 04:44:22
合計ジャッジ時間 6,638 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import math
from math import sin, cos, tan
from functools import reduce
from collections import deque
import heapq

sys.setrecursionlimit(1000000)
intm1 = lambda x: int(x) - 1

N = int(input())
A = list(map(int, input().split()))
cnt1 = cnt2 = 0
for a in A:
	if a == 1:
		cnt1 += 1
	elif a == 2:
		cnt2 += 1
print(
	cnt2 * (cnt2 - 1) // 2 * 1
	+ cnt2 * (N - cnt1 - cnt2) * 1
	+ (N - cnt1 - cnt2) * (N - cnt1 - cnt2 - 1) // 2 * 1
	+ cnt1 * (cnt1 - 1) // 2 * 2
	+ cnt1 * (N - cnt1 - cnt2) * 2
	+ cnt1 * cnt2 * 3)
0