結果

問題 No.510 二次漸化式
コンテスト
ユーザー tnakao0123
提出日時 2017-04-29 15:37:52
言語 C++11（廃止可能性あり）
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 80 ms / 3,000 ms
コード長 3,672 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,245 ms
コンパイル使用メモリ 87,404 KB
実行使用メモリ 36,492 KB
最終ジャッジ日時 2024-09-13 22:51:25
合計ジャッジ時間 5,696 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 34
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:167:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  167 |   scanf("%d%d", &n, &qn);
      |   ~~~~~^~~~~~~~~~~~~~~~~
main.cpp:177:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  177 |     scanf("%c %d", &op, &i);
      |     ~~~~~^~~~~~~~~~~~~~~~~~
main.cpp:181:12: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  181 |       scanf("%d", &v);
      |       ~~~~~^~~~~~~~~~

ソースコード

diff #
raw source code 

/* -*- coding: utf-8 -*-
 *
 * 510.cc: No.510 二次漸化式 - yukicoder
 */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_N = 100000;
const int MAX_E2 = 1 << 18; // 262144
const int N = 4;

typedef long long ll;
const ll MOD = 1000000007;

/* typedef */

typedef vector<int> vi;
typedef queue<int> qi;
typedef pair<int,int> pii;

struct Vec {
  ll v[N];
  Vec(): v() {}
  Vec(const ll _v[N]) { memcpy(v, _v, sizeof(v)); }
};

const ll IDV0[] = {1, 1, 1, 1};
const Vec V0(IDV0);

struct Mat {
  ll m[N][N];
  Mat() {}
  Mat(const ll _m[N][N]) { memcpy(m, _m, sizeof(m)); }
  void init(const ll _m[N][N]) { memcpy(m, _m, sizeof(m)); }

  Mat operator*(const Mat &a) const {
    Mat r;
    for (int i = 0; i < N; i++)
      for (int j = 0; j < N; j++) {
	r.m[i][j] = 0;
	for (int k = 0; k < N; k++)
	  r.m[i][j] += m[i][k] * a.m[k][j] % MOD;
	r.m[i][j] %= MOD;
      }
    return r;
  }

  Vec operator*(const Vec &v) const {
    Vec r;
    for (int i = 0; i < N; i++) {
      r.v[i] = 0;
      for (int j = 0; j < N; j++)
	r.v[i] += m[i][j] * v.v[j] % MOD;
      r.v[i] %= MOD;
    }
    return r;
  }

  void print() {
    for (int i = 0; i < N; i++) {
      for (int j = 0; j < N; j++) printf("%lld ", m[i][j]);
      putchar('\n');
    }
  }
};

const ll ID[N][N] = {
  {1, 0, 0, 0},
  {0, 0, 0, 1},
  {0, 0, 0, 1},
  {0, 0, 0, 1}
};

Mat IDMAT(ID);

struct SegTreeMat {
  int n, e2;
  Mat nodes[MAX_E2];
  SegTreeMat() {}

  void init(int _n) {
    n = _n;
    for (e2 = 1; e2 < n; e2 <<= 1);
    fill(nodes, nodes + MAX_E2, IDMAT);
  }

  Mat get(int i) { return nodes[e2 - 1 + i]; }

  void setx(int i, ll x) {
    int j = e2 - 1 + i;
    nodes[j].m[0][2] = x;
    j = (j - 1) / 2;

    for (;;) {
      nodes[j] = nodes[j * 2 + 2] * nodes[j * 2 + 1];
      if (j == 0) break;
      j = (j - 1) / 2;
    }
  }

  void sety(int i, ll y) {
    int j = e2 - 1 + i;
    nodes[j].m[1][1] = y;
    nodes[j].m[2][1] = 2 * y % MOD;
    nodes[j].m[2][2] = y * y % MOD;
    j = (j - 1) / 2;

    for (;;) {
      nodes[j] = nodes[j * 2 + 2] * nodes[j * 2 + 1];
      if (j == 0) break;
      j = (j - 1) / 2;
    }
  }

  Mat mul_range(int r0, int r1, int k, int i0, int i1) {
    if (r1 <= i0 || i1 <= r0) return IDMAT;
    if (r0 <= i0 && i1 <= r1) return nodes[k];

    int im = (i0 + i1) / 2;
    Mat v0 = mul_range(r0, r1, k * 2 + 1, i0, im);
    Mat v1 = mul_range(r0, r1, k * 2 + 2, im, i1);

    return v1 * v0;
  }

  Mat mul_range(int r0, int r1) {
    return mul_range(r0, r1, 0, 0, e2);
  }
};

/* global variables */

SegTreeMat st;

/* subroutines */

/* main */

// a{i+1}    |1 0   xi   0| ai
// b{i+1}  = |0 yi  0    1| bi
// b{i+1}^2  |0 2yi yi^2 1| bi^2
// 1         |0 0   0    1| 1

int main() {
  int n, qn;
  scanf("%d%d", &n, &qn);
  //printf("n=%d, qn=%d\n", n, qn);

  st.init(n);

  while (qn--) {
    while (getchar() != '\n');

    char op;
    int i;
    scanf("%c %d", &op, &i);

    if (op == 'x' || op == 'y') {
      int v;
      scanf("%d", &v);
      //printf("%c %d %d\n", op, i, v);
      if (op == 'x') st.setx(i, v);
      else st.sety(i, v);
    }
    else {
      //printf("%c %d\n", op, i);
      Mat mi = st.mul_range(0, i);
      //printf("mi=\n"), mi.print();
      Vec vv = mi * V0;
      printf("%lld\n", vv.v[0]);
    }
  }

  //for (int i = 0; i < n; i++) printf("%d:\n", i), st.get(i).print();
  return 0;
}
0