結果
問題 |
No.3176 転移迷宮 (Hard)
|
ユーザー |
|
提出日時 | 2025-06-06 05:30:57 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 303 ms / 3,000 ms |
コード長 | 1,874 bytes |
コンパイル時間 | 4,707 ms |
コンパイル使用メモリ | 255,860 KB |
実行使用メモリ | 43,036 KB |
最終ジャッジ日時 | 2025-06-06 05:31:15 |
合計ジャッジ時間 | 17,379 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 41 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using mint = atcoder::modint998244353; void solve(){ // 入力を受け取る int N, K; cin >> N >> K; // 行列を生成 // A : 係数行列 // ans : 右辺の値 const int siz = 2 * K + 1; vector A(N, vector<mint>(siz, -1)); vector<mint> ans(N, mint::raw(siz)); for(int i = 0; i < N; i++){ // 主対角成分を 2 * K にしている A[i][K] = 2 * K; // 行列の範囲外を 0 にしている (やらなくても良い) for(int j = 0; i + j < K; j++){ A[i][j] = mint::raw(0); A[N - 1 - i][2 * K - j] = mint::raw(0); } } // ガウスの消去法を実行して、前進消去を行う for(int i = 0; i < N; i++){ // 主対角成分を1に合わせる処理 mint div = A[i][K].inv(); for(int j = K; j < siz; j++) A[i][j] *= div; ans[i] *= div; // 今着目している主対角成分で他の行の列を消去しにいく for(int j = 1; j <= K && i + j < N; j++){ // 着目している主対角成分は A[i][K] // 行列を45度回転しているので同じ列は行が1増えるにつれ1減る方向にずれる mint coef = A[i + j][K - j]; for(int l = K; l < siz; l++){ A[i + j][l - j] -= coef * A[i][l]; } ans[i + j] -= coef * ans[i]; } } // 後退代入 for(int i = N - 2; i >= 0; i--){ for(int j = 1; j <= K && i + j < N; j++){ ans[i] -= A[i][K + j] * ans[i + j]; } } // 答えを出力 for(int i = 0; i < N; i++) cout << ans[i].val() << (i + 1 == N ? '\n' : ' '); } int main(){ ios::sync_with_stdio(false); cin.tie(0); int T; cin >> T; while(T--) solve(); }