結果

問題 No.243 出席番号(2)
ユーザー gew1fw
提出日時 2025-06-12 21:40:43
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 772 bytes
コンパイル時間 141 ms
コンパイル使用メモリ 82,344 KB
実行使用メモリ 77,748 KB
最終ジャッジ日時 2025-06-12 21:44:42
合計ジャッジ時間 3,378 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 6 MLE * 24
権限があれば一括ダウンロードができます

ソースコード

diff #

MOD = 10**9 + 7

n = int(input())
A = [int(input()) for _ in range(n)]

from collections import defaultdict

cnt = defaultdict(int)
for a in A:
    cnt[a] += 1

vals = list(cnt.values())
m = len(vals)
max_k = min(m, n)

# Initialize dp array for elementary symmetric sums
dp = [0] * (max_k + 1)
dp[0] = 1

for c in vals:
    for k in range(min(m, max_k), 0, -1):
        dp[k] = (dp[k] + dp[k-1] * c) % MOD

# Precompute factorials modulo MOD
fact = [1] * (n + 1)
for i in range(1, n + 1):
    fact[i] = fact[i-1] * i % MOD

result = 0
for k in range(0, max_k + 1):
    if k > n:
        continue
    ek = dp[k]
    sign = 1 if k % 2 == 0 else MOD - 1
    term = (sign * ek) % MOD
    term = term * fact[n - k] % MOD
    result = (result + term) % MOD

print(result % MOD)
0