結果
問題 | No.510 二次漸化式 |
ユーザー |
![]() |
提出日時 | 2017-05-02 20:14:00 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 952 ms / 3,000 ms |
コード長 | 3,067 bytes |
コンパイル時間 | 1,966 ms |
コンパイル使用メモリ | 182,808 KB |
実行使用メモリ | 71,936 KB |
最終ジャッジ日時 | 2024-09-14 04:45:46 |
合計ジャッジ時間 | 30,606 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 34 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define rep(i,a,b) for(int i=a;i<b;i++) #define INF 1LL<<60 typedef vector<int> Vec; typedef vector<Vec> Mat; int mod = 1000000007; int add(int x, int y) { return (x += y) >= mod ? x - mod : x; } template<class... T> int add(int x, T... y) { return add(x, add(y...)); } int mul(int x, int y) { return 1LL * x * y % mod; } template<class... T> int mul(int x, T... y) { return mul(x, mul(y...)); } int sub(int x, int y) { return add(x, mod - y); } int modpow(int a, long long b) { int ret = 1; while (b > 0) { if (b & 1) ret = 1LL * ret * a % mod; a = 1LL * a * a % mod; b >>= 1; } return ret; } int modinv(int a) { return modpow(a, mod - 2); } Vec mulMatVec(Mat a, Vec b) { int n = b.size(); Vec ret(n, 0); rep(i, 0, n) rep(j, 0, n) ret[i] = add(ret[i], mul(a[i][j], b[j])); return ret; } Mat mulMatMat(Mat a, Mat b) { int n = a.size(); Mat ret(n, Vec(n, 0)); rep(i, 0, n) rep(j, 0, n) rep(k, 0, n) ret[i][j] = add(ret[i][j], mul(a[i][k], b[k][j])); return ret; } struct func { Mat m; func(){ m = Mat(4, Vec(4, 0)); m[0][0] = 1; m[1][3] = 1; m[2][3] = 1; m[3][3] = 1; } }; func operator*(func x, func y) { func res; res.m = mulMatMat(x.m, y.m); return res; } //----------------------------------------------------------------- template<class V, int NV> struct SegTree { V comp(V l, V r) { return r * l; }; vector<V> val; SegTree() { val = vector<V>(NV * 2); } V get(int x, int y, int l = 0, int r = NV, int k = 1) { // x<=i<y if (r <= x || y <= l) return V(); if (x <= l && r <= y) return val[k]; auto A = get(x, y, l, (l + r) / 2, k * 2); auto B = get(x, y, (l + r) / 2, r, k * 2 + 1); return comp(A, B); } void update(int i, V v) { i += NV; val[i] = v; while (i>1) i >>= 1, val[i] = comp(val[i * 2], val[i * 2 + 1]); } }; SegTree<func, 1 << 17> st; //----------------------------------------------------------------------------------- int N, Q; int X[101010], Y[101010]; void make(int i) { func f; f.m[0][2] = X[i]; f.m[1][1] = Y[i]; f.m[2][1] = mul(2, Y[i]); f.m[2][2] = mul(Y[i], Y[i]); st.update(i, f); } //----------------------------------------------------------------------------------- int main() { cin.tie(0); ios::sync_with_stdio(false); cin >> N >> Q; rep(q, 0, Q) { char t; cin >> t; if (t == 'x') { int i, v; cin >> i >> v; X[i] = v; make(i); } else if (t == 'y') { int i, v; cin >> i >> v; Y[i] = v; make(i); } else { int i; cin >> i; if (i == 0) { printf("1\n"); continue; } auto f = st.get(0, i); Vec v(4, 0); v[0] = 1; v[1] = 1; v[2] = 1; v[3] = 1; v = mulMatVec(f.m, v); printf("%d\n", v[0]); } } }