結果
問題 |
No.287 場合の数
|
ユーザー |
![]() |
提出日時 | 2020-10-17 14:59:59 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 71 ms / 5,000 ms |
コード長 | 1,122 bytes |
コンパイル時間 | 229 ms |
コンパイル使用メモリ | 82,092 KB |
実行使用メモリ | 70,016 KB |
最終ジャッジ日時 | 2024-07-21 02:30:27 |
合計ジャッジ時間 | 3,022 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
import sys INF = 1 << 60 MOD = 10**9 + 7 # 998244353 sys.setrecursionlimit(2147483647) input = lambda:sys.stdin.readline().rstrip() from cmath import pi, rect def _fft(A, inverse = False): N = len(A) logN = (N - 1).bit_length() step = N for k in range(logN): step >>= 1 w = rect(1, pi / (1 << k) * (1 - 2 * inverse)) wj = 1 nA = [0] * N for j in range(1 << k): for i in range(1 << logN - k - 1): s, t = i + j * step, i + j * step + (N >> 1) ps, pt = i + j * step * 2, i + j * step * 2 + step nA[s], nA[t] = A[ps] + A[pt] * wj, A[ps] - A[pt] * wj wj *= w A = nA return A def convolution(f, g): N = 1 << (len(f) + len(g) - 2).bit_length() Ff, Fg = _fft(f + [0] * (N - len(f))), _fft(g + [0] * (N - len(g))) fg = _fft([a * b / N for a, b in zip(Ff, Fg)], inverse = True) del fg[len(f) + len(g) - 1:] return fg def resolve(): n = int(input()) f = [1] * (n + 1) for _ in range(3): f = convolution(f, f) print(round(f[6 * n].real)) resolve()