結果
問題 | No.426 往復漸化式 |
ユーザー | kimiyuki |
提出日時 | 2017-03-10 00:56:42 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 380 ms / 5,000 ms |
コード長 | 4,173 bytes |
コンパイル時間 | 1,103 ms |
コンパイル使用メモリ | 83,528 KB |
実行使用メモリ | 22,840 KB |
最終ジャッジ日時 | 2024-06-24 00:25:57 |
合計ジャッジ時間 | 7,841 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,944 KB |
testcase_02 | AC | 3 ms
6,940 KB |
testcase_03 | AC | 8 ms
6,940 KB |
testcase_04 | AC | 7 ms
6,940 KB |
testcase_05 | AC | 50 ms
6,944 KB |
testcase_06 | AC | 50 ms
6,944 KB |
testcase_07 | AC | 330 ms
22,644 KB |
testcase_08 | AC | 329 ms
22,780 KB |
testcase_09 | AC | 359 ms
22,772 KB |
testcase_10 | AC | 358 ms
22,736 KB |
testcase_11 | AC | 327 ms
22,792 KB |
testcase_12 | AC | 357 ms
22,820 KB |
testcase_13 | AC | 370 ms
22,608 KB |
testcase_14 | AC | 371 ms
22,840 KB |
testcase_15 | AC | 338 ms
22,672 KB |
testcase_16 | AC | 368 ms
22,812 KB |
testcase_17 | AC | 380 ms
22,700 KB |
testcase_18 | AC | 365 ms
22,744 KB |
testcase_19 | AC | 326 ms
22,816 KB |
testcase_20 | AC | 350 ms
22,740 KB |
testcase_21 | AC | 362 ms
22,644 KB |
testcase_22 | AC | 352 ms
22,732 KB |
ソースコード
#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; }