結果

問題 No.3172 三角関数べき乗のフーリエ級数展開
ユーザー t98slider
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

# 入力を受け取る
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)
0