結果
問題 |
No.3227 Matrix Query
|
ユーザー |
![]() |
提出日時 | 2025-08-08 22:00:21 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2,045 ms / 8,000 ms |
コード長 | 2,260 bytes |
コンパイル時間 | 5,162 ms |
コンパイル使用メモリ | 272,048 KB |
実行使用メモリ | 11,136 KB |
最終ジャッジ日時 | 2025-08-08 22:01:00 |
合計ジャッジ時間 | 37,133 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 28 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using isize = size_t; using i32 = int; using u32 = unsigned int; using i64 = long long; using u64 = unsigned long long; using i128 = __int128_t; using u128 = __uint128_t; using f64 = long double; using p2 = pair<i64, i64>; using el = tuple<i64, i64, i64>; using mint = atcoder::modint998244353; void _main(); int main() { cin.tie(0); ios::sync_with_stdio(false); _main(); } i64 pow(i64 x, i64 n) { i64 res = 1; i64 t = x; while (n > 0) { if (n & 1) { res = res * t; } t = t * t; n >>= 1; } return res; } i64 pow(i64 x, i64 n, i64 m) { i64 res = 1; i64 t = x % m; while (n > 0) { if (n & 1) { res = res * t % m; } t = t * t % m; n >>= 1; } return res; } using mat = vector<vector<i64>>; using v4 = vector<i64>; i64 K; mat mul(mat a, mat b) { assert(a[0].size() == b.size()); mat res(a.size(), v4(b[0].size(), 0)); for (i64 i = 0; i < a.size(); i++) { for (i64 j = 0; j < b[0].size(); j++) { for (i64 k = 0; k < a[0].size(); k++) { res[i][j] += a[i][k] * b[k][j]; res[i][j] = (res[i][j] % K + K) % K; } } } return res; } mat pow(mat a, i64 n) { mat t = a; assert(a.size() == a[0].size()); mat res(a.size(), v4(a.size(), 0)); for (i64 i = 0; i < a.size(); i++) res[i][i] = 1; while (n > 0) { if (n & 1) res = mul(res, t); t = mul(t, t); n >>= 1; } return res; } mat e() { return {{1, 0}, {0, 1}}; } void _main() { i64 N; cin >> K >> N; vector<mat> A(N, mat(2, v4(2))); atcoder::segtree<mat, mul, e> seg(N); for (i64 i = 0; i < N; i++) { for (i64 j = 0; j < 2; j++) { for (i64 k = 0; k < 2; k++) { cin >> A[i][j][k]; A[i][j][k] = (A[i][j][k] % K + K) % K; } } seg.set(i, A[i]); } i64 q; cin >> q; while (q--) { i64 i, l, r; cin >> i >> l >> r; mat y(2, v4(2)); for (i64 j = 0; j < 2; j++) for (i64 k = 0; k < 2; k++) cin >> y[j][k], y[j][k] = (y[j][k] % K + K) % K; i--, l--; seg.set(i, y); mat res = seg.prod(l, r); for (i64 j = 0; j < 2; j++) { for (i64 k = 0; k < 2; k++) { cout << res[j][k] << " "; } cout << "\n"; } } }