結果

問題 No.3175 転移迷宮 (Easy)
ユーザー t98slider
提出日時 2025-02-16 01:12:21
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 132 ms / 3,000 ms
コード長 1,118 bytes
コンパイル時間 3,653 ms
コンパイル使用メモリ 254,604 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2025-02-20 21:00:59
合計ジャッジ時間 8,638 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using mint = atcoder::modint998244353;

void solve(){
    int n, k;
    cin >> n >> k;
    vector A(n, valarray<mint>(n + 1));
    for(int i = 0; i < n; i++){
        for(int j = 0; j < n; j++){
            if(abs(i - j) <= k) A[i][j] = (i == j ? 2 * k : -1);
        }
    }
    for(int i = 0; i < n; i++) A[i][n] = 2 * k + 1;
    int rank = 0;
    for(int j = 0; j < n; j++){
        int pivot = -1;
        for(int i = rank; i < n; i++){
            if(A[i][j] != 0){
                pivot = i;
                break;
            }
        }
        if(pivot == -1) continue;
        swap(A[pivot], A[rank]);
        mint div = mint(1) / A[rank][j];
        A[rank] *= div;
        for(int i = 0; i < n; i++){
            if(i == rank) continue;
            if(A[i][j] != 0) A[i] -= A[i][j] * A[rank];
        }
        rank++;
    }
    for(int i = 0; i < n; i++){
        cout << A[i][n].val() << (i + 1 == n ? '\n' : ' ');
    }
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T;
    cin >> T;
    while(T--) solve();
}
0