結果
問題 |
No.3172 三角関数べき乗のフーリエ級数展開
|
ユーザー |
|
提出日時 | 2025-02-23 13:11:38 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 94 ms / 2,000 ms |
コード長 | 697 bytes |
コンパイル時間 | 443 ms |
コンパイル使用メモリ | 82,292 KB |
実行使用メモリ | 87,332 KB |
最終ジャッジ日時 | 2025-02-23 13:11:41 |
合計ジャッジ時間 | 2,539 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 15 |
ソースコード
# 入力を受け取る N = int(input()) # 階乗とその逆元を前計算 fact = [0] * (N + 1) inv = [0] * (N + 1) fact[0] = 1 for i in range(1, N + 1): fact[i] = (i * fact[i - 1]) % 998244353 inv[N] = pow(fact[N], -1, 998244353) for i in range(N, 0, -1): inv[i - 1] = (i * inv[i]) % 998244353 # 二項係数を計算する関数 def binom(r, c): if r < 0 or c < 0 or r < c: return 0 return (((fact[r] * inv[c]) % 998244353) * inv[r - c]) % 998244353 # 答えの係数を計算 ans = [0] * (N + 1) for i in range(-N, N + 1, 2): ans[abs(i)] += binom(N, (N + i) // 2) if ans[abs(i)] >= 998244353: ans[abs(i)] -= 998244353 # 答えを出力 print(*ans)