結果
問題 |
No.3175 転移迷宮 (Easy)
|
ユーザー |
|
提出日時 | 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 |
ソースコード
#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(); }