結果

問題 No.426 往復漸化式
ユーザー kimiyukikimiyuki
提出日時 2017-03-10 00:56:42
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 349 ms / 5,000 ms
コード長 4,173 bytes
コンパイル時間 875 ms
コンパイル使用メモリ 83,792 KB
実行使用メモリ 22,588 KB
最終ジャッジ日時 2023-09-06 05:44:45
合計ジャッジ時間 7,483 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 7 ms
4,376 KB
testcase_04 AC 6 ms
4,376 KB
testcase_05 AC 47 ms
5,572 KB
testcase_06 AC 46 ms
5,372 KB
testcase_07 AC 300 ms
22,408 KB
testcase_08 AC 303 ms
22,516 KB
testcase_09 AC 337 ms
22,436 KB
testcase_10 AC 338 ms
22,476 KB
testcase_11 AC 303 ms
22,408 KB
testcase_12 AC 330 ms
22,540 KB
testcase_13 AC 349 ms
22,556 KB
testcase_14 AC 332 ms
22,552 KB
testcase_15 AC 304 ms
22,552 KB
testcase_16 AC 334 ms
22,408 KB
testcase_17 AC 345 ms
22,448 KB
testcase_18 AC 326 ms
22,532 KB
testcase_19 AC 305 ms
22,588 KB
testcase_20 AC 316 ms
22,484 KB
testcase_21 AC 327 ms
22,516 KB
testcase_22 AC 323 ms
22,460 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <array>
#include <functional>
#define repeat(i,n) for (int i = 0; (i) < int(n); ++(i))
using ll = long long;
using namespace std;

const int mod = 1e9+7;

template <typename T, size_t H, size_t W>
using matrix = array<array<T,W>,H>;
template <typename T, size_t A, size_t B, size_t C>
matrix<T,A,C> operator * (matrix<T,A,B> const & a, matrix<T,B,C> const & b) {
    matrix<T,A,C> c = {};
    repeat (y,A) {
        repeat (z,B) {
            repeat (x,C) {
                c[y][x] = (c[y][x] + a[y][z] *(ll) b[z][x]) % mod;
            }
        }
    }
    return c;
}
template <typename T, size_t H, size_t W>
array<T,H> operator * (matrix<T,H,W> const & a, array<T,W> const & b) {
    array<T,H> c = {};
    repeat (y,H) {
        repeat (z,W) {
            c[y] = (c[y] + a[y][z] *(ll) b[z]) % mod;
        }
    }
    return c;
}
template <typename T, size_t H, size_t W>
matrix<T,H,W> operator + (matrix<T,H,W> const & a, matrix<T,H,W> const & b) {
    matrix<T,H,W> c;
    repeat (y,H) {
        repeat (x,W) {
            c[y][x] = (a[y][x] + b[y][x]) % mod;
        }
    }
    return c;
}
template <typename T, size_t N>
array<T,N> operator + (array<T,N> const & a, array<T,N> const & b) {
    array<T,N> c;
    repeat (i,N) {
        c[i] = (a[i] + b[i]) % mod;
    }
    return c;
}
template <typename T, size_t H, size_t W>
matrix<T,H,W> matrix_zero() { return {}; }
template <typename T, size_t N>
matrix<T,N,N> matrix_unit() { matrix<T,N,N> a = {}; repeat (i,N) a[i][i] = 1; return a; }

template <typename T>
struct segment_tree { // on monoid
    int n;
    vector<T> a;
    function<T (T,T)> append; // associative
    T unit; // unit
    segment_tree() = default;
    segment_tree(int a_n, T a_unit, function<T (T,T)> a_append) {
        n = 1; while (n < a_n) n *= 2;
        a.resize(2*n-1, a_unit);
        unit = a_unit;
        append = a_append;
    }
    template <typename F>
    void point_update(int i, F func) { // 0-based
        a[i+n-1] = func(a[i+n-1]);
        for (i = (i+n)/2; i > 0; i /= 2) {
            a[i-1] = append(a[2*i-1], a[2*i]);
        }
    }
    T point_get(int i) {
        return range_concat(i, i+1);
    }
    T range_concat(int l, int r) { // 0-based, [l, r)
        T lacc = unit, racc = unit;
        for (l += n, r += n; l < r; l /= 2, r /= 2) {
            if (l % 2 == 1) lacc = append(lacc, a[(l ++) - 1]);
            if (r % 2 == 1) racc = append(a[(-- r) - 1], racc);
        }
        return append(lacc, racc);
    }
};

struct range_t {
    matrix<int,3,3> A;
    matrix<int,2,2> B;
    matrix<int,2,3> S;
};

int main() {
    int n; cin >> n;
    array<int,3> a0; repeat (i,3) cin >> a0[i];
    array<int,2> bn; repeat (i,2) cin >> bn[i];
    segment_tree<range_t> segtree(n+1, (range_t) { matrix_unit<int,3>(), matrix_unit<int,2>(), matrix_zero<int,2,3>() }, [&](range_t const & l, range_t const & r) {
        range_t m;
        m.A = r.A * l.A;
        m.B = l.B * r.B;
        m.S = l.S + l.B * r.S * l.A;
        return m;
    });
    repeat (i,n+1) {
        segtree.point_update(i, [&](range_t m) {
            repeat (y,2) repeat (x,3) m.S[y][x] = 6*i + 3*y+x;
            return m;
        });
    }
    int query; cin >> query;
    while (query --) {
        string type; int i; cin >> type >> i;
        if (type == "a") {
            segtree.point_update(i, [](range_t m) {
                repeat (y,3) repeat (x,3) cin >> m.A[y][x];
                return m;
            });
        } else if (type == "b") {
            segtree.point_update(i, [](range_t m) {
                repeat (y,2) repeat (x,2) cin >> m.B[y][x];
                return m;
            });
        } else if (type == "ga") {
            range_t m = segtree.range_concat(0, i);
            array<int,3> ai = m.A * a0;
            cout << ai[0] << ' ' << ai[1] << ' ' << ai[2] << endl;
        } else if (type == "gb") {
            range_t l = segtree.range_concat(0, i+1);
            range_t r = segtree.range_concat(i+1, n+1);
            array<int,2> bi = r.B * bn + r.S * l.A * a0;
            cout << bi[0] << ' ' << bi[1] << endl;
        }
    }
    return 0;
}
0