結果

問題 No.3777 Train Seats
ユーザー titia
提出日時 2026-11-12 02:29:06
言語 PyPy3
(8.0.1 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 54 ms / 2,000 ms
+ 448µs
コード長 774 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 87 ms
コンパイル使用メモリ 64,404 KB
実行使用メモリ 66,232 KB
最終ジャッジ日時 2026-11-12 02:29:21
合計ジャッジ時間 5,946 ms
ジャッジサーバーID
(参考情報)
judge5_0 / judge3_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

# 解説を見た。これは思いつけなきゃダメ!
# 使う椅子がどれか? を決めれば、順序が求められることを見抜く。

import sys
input = sys.stdin.readline

N=int(input())

mod=998244353

F=[1]
for i in range(1,N+1):
    F.append(F[-1]*i%mod)

DP=[[0]*N for i in range(N+1)]
DP[1][0]=1
if N>=2:
    DP[1][1]=1

for i in range(N):
    for j in range(1,N+1):
        if DP[j][i]==0:
            continue

        if i+2<N:
            DP[j+1][i+2]=(DP[j+1][i+2]+DP[j][i])%mod

        if i+3<N:
            DP[j+1][i+3]=(DP[j+1][i+3]+DP[j][i])%mod

ANS=0
for i in range(N+1):
    if N>=2:
        ANS+=(DP[i][-1]+DP[i][-2])*F[i]*F[N-i]%mod
        ANS%=mod
    else:
        ANS+=(DP[i][-1])*F[i]*F[N-i]%mod
        ANS%=mod

print(ANS)
0