結果
問題 |
No.3175 転移迷宮 (Easy)
|
ユーザー |
|
提出日時 | 2025-02-18 02:13:06 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 143 ms / 3,000 ms |
コード長 | 1,567 bytes |
コンパイル時間 | 4,618 ms |
コンパイル使用メモリ | 253,348 KB |
実行使用メモリ | 6,824 KB |
最終ジャッジ日時 | 2025-02-20 21:00:59 |
合計ジャッジ時間 | 9,643 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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; // 行列を生成 (拡大係数行列として扱うために N x (N+1) にしています) 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++) { #if 1 // 一応いれていますが、今回の問題に限って言えばpivotの処理は不要です。 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]); #endif // A[rank][j] を 1 にする処理 mint div = A[rank][j].inv(); A[rank] *= div; // A[i][j] が非ゼロの成分を消去しにいく 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(); }