結果
問題 | No.243 出席番号(2) |
ユーザー |
![]() |
提出日時 | 2016-04-23 20:24:43 |
言語 | Python2 (2.7.18) |
結果 |
AC
|
実行時間 | 70 ms / 2,000 ms |
コード長 | 1,686 bytes |
コンパイル時間 | 145 ms |
コンパイル使用メモリ | 6,912 KB |
実行使用メモリ | 6,912 KB |
最終ジャッジ日時 | 2024-10-04 15:22:48 |
合計ジャッジ時間 | 1,550 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
def ilog2(n):return 0 if n <= 0 else n.bit_length() - 1def pack(pack, shamt):size = len(pack)while size > 1:npack = []for i in range(0, size - 1, 2):npack += [pack[i] | (pack[i+1] << shamt)]if size & 1:npack += [pack[-1]]pack, size, shamt = npack, (size + 1) >> 1, shamt << 1return pack[0]def unpack(M, size, shamt):s, sizes = size, []while s > 1:sizes += [s]s = (s + 1) >> 1ret = [M]for size in sizes[::-1]:mask, nret = (1 << shamt) - 1, []for c in ret:nret += [c & mask, c >> shamt]ret, shamt = nret[:size], shamt >> 1return retdef poly_mul_mod(f, g, mod):size = min(len(f), len(g))shift = ((mod - 1) ** 2 * size).bit_length()rsize = len(f) + len(g) - 1h = unpack(pack(f, shift) * pack(g, shift), rsize, shift * (1 << ilog2(rsize - 1)))return [int(x % mod) for x in h]def poly_coefficients_mod(roots, mod):def rec(beg, end):if end - beg == 1:return [1, -roots[beg] % mod]else:mid = (beg + end) // 2return poly_mul_mod(rec(beg, mid), rec(mid, end), mod)if not roots:return [1]ret = [0] * (len(roots) + 1)return rec(0, len(roots))def prob243():from sys import stdinN = int(stdin.readline())MOD = 10 ** 9 + 7cnts = [0] * 5000for line in stdin:a = int(line)cnts[a] += 1roots = []for i in range(N):if cnts[i]:roots += [-cnts[i]]poly = poly_coefficients_mod(roots, MOD)facts = [1]for i in range(1, N + 1):facts += [facts[-1] * i % MOD]ans = 0for i in range(len(poly)):ans += (-1) ** i * poly[i] * facts[-1 - i] % MODprint(ans % MOD)prob243()