結果
問題 |
No.3175 転移迷宮 (Easy)
|
ユーザー |
|
提出日時 | 2025-02-18 02:17:33 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 286 ms / 3,000 ms |
コード長 | 1,215 bytes |
コンパイル時間 | 299 ms |
コンパイル使用メモリ | 82,508 KB |
実行使用メモリ | 78,920 KB |
最終ジャッジ日時 | 2025-02-20 21:01:20 |
合計ジャッジ時間 | 10,347 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 44 |
ソースコード
T = int(input()) for testcase in range(T): # 入力を受け取る N, K = map(int, input().split()) # 行列を生成 (拡大係数行列として扱うために N x (N+1) にしています) A = [[0 for j in range(N + 1)] for i in range(N)] for i in range(N): A[i][N] = 2 * K + 1 for j in range(N): if abs(i - j) <= K: A[i][j] = 2 * K if i == j else 998244353 - 1 # ガウスの消去法を実行 for rank in range(N): # (注意) 今回の問題に限って言えばpivotの処理は不要なためサボっています # A[rank][rank] を 1 にする処理 div = pow(A[rank][rank], -1, 998244353) for j in range(rank, N + 1, 1): A[rank][j] *= div A[rank][j] %= 998244353 # A[i][rank] において非ゼロの成分を削除する処理 for i in range(N): if i == rank or A[i][rank] == 0: continue coef = A[i][rank] for j in range(rank, N + 1, 1): A[i][j] -= coef * A[rank][j] A[i][j] %= 998244353 # 答えを出力 print(*[A[i][N] for i in range(N)])