結果

問題 No.510 二次漸化式
コンテスト
ユーザー tossy
提出日時 2017-05-06 19:35:14
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 360 ms / 3,000 ms
コード長 2,855 bytes
コンパイル時間 2,366 ms
コンパイル使用メモリ 182,236 KB
実行使用メモリ 89,012 KB
最終ジャッジ日時 2024-09-14 13:59:53
合計ジャッジ時間 12,472 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define repl(i,a,b) for(int i=(int)(a);i<(int)(b);i++)
#define rep(i,n) repl(i,0,n)
#define mp(a,b) make_pair((a),(b))
#define pb(a) push_back((a))
#define all(x) (x).begin(),(x).end()
#define uniq(x) sort(all(x)),(x).erase(unique(all(x)),end(x))
#define fi first
#define se second
#define dbg(x) cout<<#x" = "<<((x))<<endl
template<class T,class U> ostream& operator<<(ostream& o, const pair<T,U> &p){o<<"("<<p.fi<<","<<p.se<<")";return o;}
template<class T> ostream& operator<<(ostream& o, const vector<T> &v){o<<"[";for(T t:v){o<<t<<",";}o<<"]";return o;}

#define INF 2147483600
#define MOD 1000000007

typedef vector<vector<long>> mat;
mat mat_mul(const mat &A, const mat &B){
  int n=A.size(), m=B[0].size(), l=B.size();
  mat ret(n, vector<long>(m, 0));
  rep(i,n) rep(k,l) if(A[i][k]!=0) rep(j,m){
    (ret[i][j] += A[i][k] * B[k][j]) %= MOD;
  }
  return ret;
}

mat newmat(long x, long y){
  return mat({{1,0,x,0},{0,y,0,1},{0,2*y,y*y%MOD,1},{0,0,0,1}});
}

class SegTree {
public:
  int n;
  vector<mat> data;
  mat ei;

  SegTree(int n_){
    n=1;
    while(n<n_) n*=2;
    data.resize(2*n-1);

    ei = mat(4, vector<long>(4,0));
    rep(i,4) ei[i][i] = 1;

    mat zeromat = newmat(0,0);
    repl(i,n-1, 2*n-1) data[i] = zeromat;
    for(int i=n-2; i>=0; i--) data[i] = mat_mul(data[i*2+2], data[i*2+1]);
  }

  mat _query(int a, int b, int k, int l, int r){
    if(r<=a || b<=l) return ei;
    if(a<=l && r<=b) return data[k];
    mat vl=_query(a,b,k*2+1,l,(l+r)/2);
    mat vr=_query(a,b,k*2+2,(l+r)/2,r);
    return mat_mul(vr, vl);
  }
  mat query(int a, int b){return _query(a,b,0,0,n);}

  void update(int k, int x, int y){
    k += n-1;
    data[k] = newmat(x,y);
    while(k>0){
      k = (k-1)/2;
      data[k] = mat_mul(data[k*2+2], data[k*2+1]);
    }
  }
};

int main(){
  int n,q;
  cin>>n>>q;
  vector<long> x(n,0), y(n,0);
  SegTree st(n);

  rep(_,q){
    char c; scanf(" %c", &c);
    if(c=='x'){
      int i,v;
      cin>>i>>v;
      st.update(i, v, y[i]);
      x[i] = v;
    } else if(c=='y'){
      int i,v;
      cin>>i>>v;
      st.update(i, x[i], v);
      y[i] = v;
    } else if(c=='a'){
      int i;
      cin>>i;
      mat m = st.query(0,i);
      long ans = m[0][0] + m[0][1] + m[0][2] + m[0][3];
      cout << ans%MOD << endl;
    } else {
      assert(false);
    }
    // cout << "a:";
    // rep(i,n+1){
    //   mat m = st.query(0,i);
    //   long ans = m[0][0] + m[0][1] + m[0][2] + m[0][3];
    //   cout << ans << ",";
    // }
    // cout << endl;
    // cout << "b:";
    // rep(i,n+1){
    //   mat m = st.query(0,i);
    //   long ans = m[1][0] + m[1][1] + m[1][2] + m[1][3];
    //   cout << ans << ",";
    //   long bb = m[2][0] + m[2][1] + m[2][2] + m[2][3];
    //   assert(ans*ans%MOD == bb %MOD);
    // }
    // cout << endl;
  }

  return 0;
}
0