結果
問題 |
No.3227 Matrix Query
|
ユーザー |
|
提出日時 | 2025-08-08 21:54:56 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2,029 ms / 8,000 ms |
コード長 | 4,121 bytes |
コンパイル時間 | 2,853 ms |
コンパイル使用メモリ | 219,464 KB |
実行使用メモリ | 8,064 KB |
最終ジャッジ日時 | 2025-08-08 21:55:33 |
合計ジャッジ時間 | 34,536 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 28 |
ソースコード
#include<bits/stdc++.h> using namespace std; //#pragma GCC optimize("Ofast") #define rep(i,n) for(ll i=0;i<n;i++) #define repl(i,l,r) for(ll i=(l);i<(r);i++) #define per(i,n) for(ll i=(n)-1;i>=0;i--) #define perl(i,r,l) for(ll i=r-1;i>=l;i--) #define fi first #define se second #define pb push_back #define ins insert #define pqueue(x) priority_queue<x,vector<x>,greater<x>> #define all(x) (x).begin(),(x).end() #define CST(x) cout<<fixed<<setprecision(x) #define vtpl(x,y,z) vector<tuple<x,y,z>> #define rev(x) reverse(x); using ll=long long; using vl=vector<ll>; using vvl=vector<vector<ll>>; using pl=pair<ll,ll>; using vpl=vector<pl>; using vvpl=vector<vpl>; const ll MOD=1000000007; const ll MOD9=998244353; const int inf=1e9+10; const ll INF=4e18; const ll dy[9]={1,0,-1,0,1,1,-1,-1,0}; const ll dx[9]={0,1,0,-1,1,-1,1,-1,0}; template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; } template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; } //O(N^3logk) //https://yukicoder.me/submissions/733933 //https://atcoder.jp/contests/abc236/submissions/28778130 using vec=vector<ll>; using matrix=vector<vector<ll>>; matrix matmul(matrix &left,matrix &right,ll K){ int n=left.size(); matrix ret(n,vec(n));//+の単位元で埋める for(int i=0;i<n;i++){ for(int k=0;k<n;k++){//ループ順序の最適化 for(int j=0;j<n;j++){ ret[i][j]+=left[i][k]*right[k][j]; ret[i][j]%=K; } } } return ret; } //0-indexed,2冪のセグメントツリー template <class T> struct SegTree { private: int n;// 葉の数 vector<T> data;// データを格納するvector T def; // 初期値かつ単位元 function<T(T, T)> operation; // 区間クエリで使う処理 function<T(T, T)> change;// 点更新で使う処理 T find(int a, int b) { T val_left = def, val_right = def; for (a += (n - 1), b += (n - 1); a < b; a >>= 1, b >>= 1) { if ((a & 1) == 0){ val_left = operation(val_left, data[a]); } if ((b & 1) == 0){ val_right = operation(data[--b],val_right); } } return operation(val_left, val_right); } public: // _n:必要サイズ, _def:初期値かつ単位元, _operation:クエリ関数, // _change:更新関数 SegTree(size_t _n, T _def, function<T(T, T)> _operation, function<T(T, T)> _change=[](T a,T b){return b;}) : def(_def), operation(_operation), change(_change) { n = 1; while (n < _n) { n *= 2; } data = vector<T>(2 * n - 1, def); } void set(int i, T x) { data[i + n - 1] = x; } void build() { for (int k=n-2;k>=0;k--) data[k] = operation(data[2*k+1],data[2*k+2]); } // 場所i(0-indexed)の値をxで更新 void update(int i, T x) { i += n - 1; data[i] = change(data[i], x); while (i > 0) { i = (i - 1) / 2; data[i] = operation(data[i * 2 + 1], data[i * 2 + 2]); } } T all_prod(){ return data[0]; } // [a, b)の区間クエリを実行 T query(int a, int b) { //return _query(a, b, 0, 0, n); return find(a,b); } // 添字でアクセス T operator[](int i) { return data[i + n - 1]; } }; int main(){ ll K,n;cin >> K >> n; vvl d={{1,0},{0,1}}; SegTree<vvl> st(n,d,[&](vvl a,vvl b){return matmul(a,b,K);}); rep(i,n){ vvl a(2,vl(2)); rep(j,2)rep(k,2){ cin >> a[j][k]; a[j][k]+=K*inf; a[j][k]%=K; } st.set(i,a); } st.build(); ll q;cin >> q; while(q--){ ll i,l,r;cin >> i >> l >> r;l--;i--; vvl a(2,vl(2)); rep(j,2)rep(k,2){ cin >> a[j][k]; a[j][k]+=K*inf; a[j][k]%=K; } st.update(i,a); auto p=st.query(l,r); cout << p[0][0] <<" " << p[0][1] << endl; cout << p[1][0] <<" " << p[1][1] << endl; } }