結果
問題 | No.900 aδδitivee |
ユーザー | ecottea |
提出日時 | 2021-08-12 02:10:46 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 381 ms / 2,000 ms |
コード長 | 19,972 bytes |
コンパイル時間 | 2,423 ms |
コンパイル使用メモリ | 189,584 KB |
実行使用メモリ | 41,080 KB |
最終ジャッジ日時 | 2024-09-25 11:54:44 |
合計ジャッジ時間 | 12,527 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,376 KB |
testcase_02 | AC | 2 ms
5,376 KB |
testcase_03 | AC | 2 ms
5,376 KB |
testcase_04 | AC | 2 ms
5,376 KB |
testcase_05 | AC | 2 ms
5,376 KB |
testcase_06 | AC | 2 ms
5,376 KB |
testcase_07 | AC | 373 ms
35,184 KB |
testcase_08 | AC | 370 ms
35,140 KB |
testcase_09 | AC | 377 ms
35,188 KB |
testcase_10 | AC | 366 ms
35,180 KB |
testcase_11 | AC | 381 ms
35,312 KB |
testcase_12 | AC | 377 ms
35,252 KB |
testcase_13 | AC | 372 ms
35,316 KB |
testcase_14 | AC | 377 ms
35,148 KB |
testcase_15 | AC | 357 ms
35,244 KB |
testcase_16 | AC | 367 ms
35,264 KB |
testcase_17 | AC | 364 ms
35,268 KB |
testcase_18 | AC | 381 ms
35,064 KB |
testcase_19 | AC | 378 ms
35,200 KB |
testcase_20 | AC | 364 ms
35,312 KB |
testcase_21 | AC | 360 ms
35,268 KB |
testcase_22 | AC | 310 ms
41,028 KB |
testcase_23 | AC | 312 ms
41,076 KB |
testcase_24 | AC | 322 ms
41,012 KB |
testcase_25 | AC | 316 ms
40,944 KB |
testcase_26 | AC | 326 ms
41,040 KB |
testcase_27 | AC | 327 ms
41,040 KB |
testcase_28 | AC | 324 ms
41,080 KB |
ソースコード
#ifndef HIDDEN_IN_VISUAL_STUDIO // 無意味.折りたたむのが目的. // 警告の抑制 #define _CRT_SECURE_NO_WARNINGS // 使えるライブラリの読み込み #include <bits/stdc++.h> #include <functional> // function using namespace std; // 型名の短縮 using ll = long long; // -2^63 ~ 2^63 = 9 * 10^18 using ull = unsigned long long; // 0 ~ 2^64 = 1.8 * 10^19 using uint = unsigned int; // 0 ~ 2^32 = 4 * 10^9 using pii = pair<int, int>; using pll = pair<ll, ll>; using pil = pair<int, ll>; using pli = pair<ll, int>; using vi = vector<int>; using vvi = vector<vi>; using vvvi = vector<vvi>; using vll = vector<ll>; using vvll = vector<vll>; using vvvll = vector<vvll>; using vb = vector<bool>; using vvb = vector<vb>; using vc = vector<char>; using vvc = vector<vc>; // 定数の定義 const double PI = 3.141592653589793238462643383279; // 円周率 const double DEG = PI / 180.0; // θ [deg] = θ * DEG [rad] const vector<int> dx4 = { 1, 0, -1, 0 }; // 4 近傍(下,右,上,左) const vector<int> dy4 = { 0, 1, 0, -1 }; const vector<int> dx8 = { 1, 1, 0, -1, -1, -1, 0, 1 }; // 8 近傍 const vector<int> dy8 = { 0, 1, 1, 1, 0, -1, -1, -1 }; const ll INFL = (ll)1e18; const int INF = (int)1e9; const double EPS = 1e-10; // 許容誤差に応じて調整 // 汎用マクロの定義 #define all(a) (a).begin(), (a).end() #define sz(x) ((int)(x).size()) #define rep(i, n) for(int i = 0, i##_len = int(n); i < i##_len; ++i) // 0 から n-1 まで昇順 #define repi(i, s, t) for(int i = int(s), i##_end = int(t); i <= i##_end; ++i) // s から t まで昇順 #define repir(i, s, t) for(int i = int(s), i##_end = int(t); i >= i##_end; --i) // s から t まで降順 #define repe(i, a) for(const auto& i : (a)) // a の全要素 #define repb(set, d) for(int set = 0; set < (1 << int(d)); ++set) // d ビット全探索(昇順) #define repbm(mid, set, d) for(int mid = set; mid < (1 << int(d)); mid = (mid + 1) | set) // set を含む部分集合の全探索(昇順) #define repbs(sub, set) for (int sub = set, bsub = 1; bsub > 0; bsub = sub, sub = (sub - 1) & set) // set の部分集合の全探索(降順) #define repbc(set, k, d) for (int set = (1 << k) - 1, lb, nx; set < (1 << n); lb = set & -set, nx = set + lb, set = (((set & ~nx) / lb) >> 1) | nx) // 大きさ k の部分集合の全探索 #define repp(a) sort(all(a)); for(bool a##_perm = true; a##_perm; a##_perm = next_permutation(all(a))) // a の順列全て(昇順) #define repit(it, a) for(auto it = (a).begin(); it != (a).end(); ++it) // イテレータを回す(昇順) #define repitr(it, a) for(auto it = (a).rbegin(); it != (a).rend(); ++it) // イテレータを回す(降順) #define Yes(b) if(b){cout << "Yes" << endl;}else{cout << "No" << endl;} // 汎用関数の定義 inline ll pow(ll n, int k) { ll v = 1; rep(i, k) v *= n; return v; } // 工夫が必要なほど k が大きかったらどうせオーバーフローするからこれでいい inline ll pow(int n, int k) { ll v = 1; rep(i, k) v *= n; return v; } template <class T> inline bool chmax(T& M, const T& x) { if (M < x) { M = x; return true; } return false; } // 最大値を更新(更新されたら true を返す) template <class T> inline bool chmin(T& m, const T& x) { if (m > x) { m = x; return true; } return false; } // 最小値を更新(更新されたら true を返す) // 入出力用の >>, << のオーバーロード template <class T, class U> inline istream& operator>> (istream& is, pair<T, U>& p) { is >> p.first >> p.second; return is; } // pair の入力用 template <class T, class U> inline ostream& operator<< (ostream& os, const pair<T, U>& p) { os << "(" << p.first << "," << p.second << ")"; return os; } // pair の出力用 template <class T, class U, class V> inline istream& operator>> (istream& is, tuple<T, U, V>& t) { is >> get<0>(t) >> get<1>(t) >> get<2>(t); return is; } // tuple の入力用 template <class T, class U, class V> inline ostream& operator<< (ostream& os, const tuple<T, U, V>& t) { os << "(" << get<0>(t) << "," << get<1>(t) << "," << get<2>(t) << ")"; return os; } // tuple の出力用 template <class T, class U, class V, class W> inline istream& operator>> (istream& is, tuple<T, U, V, W>& t) { is >> get<0>(t) >> get<1>(t) >> get<2>(t) >> get<3>(t); return is; } // tuple の入力用 template <class T, class U, class V, class W> inline ostream& operator<< (ostream& os, const tuple<T, U, V, W>& t) { os << "(" << get<0>(t) << "," << get<1>(t) << "," << get<2>(t) << "," << get<3>(t) << ")"; return os; } // tuple の出力用 template <class T> inline istream& operator>> (istream& is, vector<T>& v) { rep(i, sz(v)) is >> v[i]; return is; } // vector の入力用 template <class T> inline ostream& operator<< (ostream& os, const vector<T>& v) { rep(i, sz(v)) os << v[i] << " "; return os; } // vector の出力用 template <class T> inline ostream& operator<< (ostream& os, const set<T>& s) { repe(x, s) os << x << " "; return os; } // set の出力用 // 手元環境(Visual Studio) #ifdef _MSC_VER #define popcount (int)__popcnt // 全ビットにおける 1 の個数 #define popcountll (int)__popcnt64 inline int ctz(uint n) { unsigned long i; _BitScanForward(&i, n); return i; } // 下位ビットに並ぶ 0 の個数 ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; } // 最大公約数 #define dump(x) cerr << "[DEBUG] " << endl << x << endl; // デバッグ出力用 #define dumpel(v) cerr << "[DEBUG]" << endl; for (const auto& x : v) {cout << x << endl;} // 提出用(GCC) #else #define popcount (int)__builtin_popcount #define popcountll (int)__builtin_popcountll #define ctz __builtin_ctz #define gcd __gcd #define dump(x) #define dumpel(v) #endif #endif // 無意味.折りたたむのが目的. ////-----------------AtCoder 専用----------------- //#include <atcoder/all> //using namespace atcoder; // //// mint で使いたい法によってここを切り替える //using mint = modint1000000007; ////using mint = modint998244353; ////using mint = modint; // modint::set_mod(10000); // mint の法の指定 // //istream& operator>> (istream& is, mint& x) { ll tmp; is >> tmp; x = tmp; return is; } // mint の入力用 //ostream& operator<< (ostream& os, const mint& x) { os << x.val(); return os; } // mint の出力用 //using vm = vector<mint>; using vvm = vector<vm>; using vvvm = vector<vvm>; ////---------------------------------------------- //【コスト付きグラフの辺】 /* * to : 行き先の頂点番号 * cost : 辺のコスト */ struct Edge { // 参考:https://nyaannyaan.github.io/library/graph/graph-template.hpp int to; // 行き先の頂点番号 ll cost; // 辺のコスト // 出力 friend ostream& operator<<(ostream& os, const Edge& e) { os << '(' << e.to << ',' << e.cost << ')'; return os; } // コストなしグラフで呼ばれたとき用 operator int() const { return to; } }; //【コスト付きグラフ】 /* * WGraph g * g[v] : 頂点 v から出る辺を並べたリスト */ using WGraph = vector<vector<Edge>>; //【コスト付きグラフの入力】O(|E|) /* * 入力を受け取り n 頂点 m 辺のコスト付きグラフを構成する. * * n : グラフの頂点の数 * m : グラフの辺の数 * g : ここにグラフを構築して返す * directed : 有向グラフなら true * one_indexed : 入力が 1-indexed で与えられるなら true */ void read_graph(int n, int m, WGraph& g, bool directed = false, bool one_indexed = true) { g = WGraph(n); rep(i, m) { int a, b; ll c; cin >> a >> b >> c; if (one_indexed) { a--; b--; } g[a].push_back({ b, c }); if (!directed) { g[b].push_back({ a, c }); } } } //【コスト付き根付き木のノード】 /* * parent : 親の頂点(なければ -1) * child : 子への辺のリスト(なければ空リスト) * depth : 深さ(根からのパスのコスト) * weight : 重さ(部分木のもつ辺の数) */ struct WTNode { int parent = -1; // 親(なければ -1) vector<Edge> child; // 子への辺(なければ空リスト) int depth = -1; // 深さ(根からのパスの長さ) int weight = -1; // 重さ(部分木のもつ辺の数) // 出力 friend ostream& operator<<(ostream& os, const WTNode& v) { os << "(p:" << v.parent << ", c:" << v.child << ", d:" << v.depth << ", w:" << v.weight << ")"; return os; } }; //【コスト付き根付き木】 /* * rt[i] : 根付き木の i 番目のノードの情報 * r : 根の頂点番号 * * WRTree(g, r) : O(|V|) * コスト付き木 g を r を根とみなしたコスト付き根付き木として受け取る. */ struct WRTree { int n; vector<WTNode> v; int r; // コンストラクタ(木と根で初期化) WRTree(WGraph& g, int r_) : n(sz(g)), v(n), r(r_) { // 再帰用の関数 // s : 注目ノード,p : s の親 function<void(int, int)> dfs = [&](int s, int p) { v[s].parent = p; v[s].child.clear(); v[s].weight = 0; repe(t, g[s]) { if (t == p) { continue; } v[t].depth = v[s].depth + 1; dfs(t, s); v[s].child.push_back(t); v[s].weight += v[t].weight + 1; } return; }; // 根 r を始点として再帰関数を呼び出す. v[r].depth = 0; dfs(r, -1); } }; //【辺のコストの取得】 /* * コスト付きグラフの辺なら辺のコストを返す. * さもなくば INFL を返す. */ inline ll get_cost(const Edge& e) { return e.cost; } inline ll get_cost(const int& t) { return INFL; } //【遅延評価セグメント木:区間加算/区間総和クエリ】 /* * range_add_sum_query(n) : O(n) * 要素数 n かつ初期値 0 で初期化する. * * range_add_sum_query(a) : O(n) * 配列 a で初期化する. * * add(i, val) : O(log n) * i 番目の要素に val を加算する. * * add(l, r, val) : O(log n) * 半開区間 [l, r) の要素に val を加算する. * * get(i, val) : O(log n) * i 番目の要素を返す. * * sum(l, r) : O(log n) * 半開区間 [l, r) の要素の総和を返す. */ template <class T> struct range_add_sum_query { // 参考:https://algo-logic.info/segment-tree/ // 完全二分木の葉の数(必ず 2 冪) int n; // 完全二分木を実現する大きさ 2 * n の配列 // 根は v[1] で,v[i] の親は v[i / 2],子は v[2 * i], v[2 * i + 1]. // 0-indexed での i 番目のデータは葉である v[i + n] に入っている. vector<T> v; // 遅延評価用の完全二分木 vector<T> lazy; // コンストラクタ(初期化なし) range_add_sum_query() {} // コンストラクタ(0 で初期化) range_add_sum_query(int n_) { // 要素数以上となる最小の 2 冪を求め,n とする. int pow2 = 1; while (pow2 < n_) { pow2 *= 2; } n = pow2; // 完全二分木を実現する大きさ 2 * n の配列を確保する. v = vector<T>(2 * n, 0); lazy = vector<T>(2 * n, 0); } // コンストラクタ(配列で初期化) range_add_sum_query(const vector<T>& a) { int n_ = sz(a); // 要素数以上となる最小の 2 冪を求め,n とする. int pow2 = 1; while (pow2 < n_) { pow2 *= 2; } n = pow2; // 完全二分木を実現する大きさ 2 * n の配列を確保する. v = vector<T>(2 * n, 0); lazy = vector<T>(2 * n, 0); // 配列の値を直接書き込む. rep(i, n_) { v[i + n] = a[i]; } // 総和をとりつつ根に向かって更新していく. repir(i, n - 1, 1) { v[i] = v[2 * i] + v[2 * i + 1]; } } // 出力 friend ostream& operator<<(ostream& os, range_add_sum_query rasq) { rep(i, rasq.n) { os << rasq.get(i) << " "; } return os; } // 遅延させていた評価を行う. void eval(int k) { // 遅延させていた評価がなければ何もしない. if (lazy[k] == 0) { return; } // 葉でなければ子に伝搬する. // 子の受ける影響は親の受ける影響の半分になる. if (k < n) { lazy[k * 2] += lazy[k] / 2; lazy[k * 2 + 1] += lazy[k] / 2; } // 自身を評価する. v[k] += lazy[k]; lazy[k] = 0; } // 一点加算:O(log N) // i 番目の要素に val を加算する. void add(int i, T val) { add(i, i + 1, val); } // 区間加算:O(log N) // 半開区間 [l, r) の要素に val を加算する. void add(int l, int r, T val) { add_rf(l, r, val, 1, 0, n, n); } // add() を実現する実際の再帰関数 // k : 注目ノード // [kl, kr) : ノード v[k] が表す区間 // num_leaf : 部分木 k の葉の数 void add_rf(int l, int r, T val, int k, int kl, int kr, int num_leaf) { // まず自身の評価を行っておく. eval(k); // 範囲外なら何もしない. if (kr <= l || r <= kl) { return; } // 完全に範囲内なら自身の値を更新する. // 葉の数の分だけ自身の値は影響を受ける. if (l <= kl && kr <= r) { lazy[k] += val * num_leaf; eval(k); return; } // 一部の範囲のみを含むなら子を見に行く. add_rf(l, r, val, k * 2, kl, (kl + kr) / 2, num_leaf / 2); add_rf(l, r, val, k * 2 + 1, (kl + kr) / 2, kr, num_leaf / 2); v[k] = v[k * 2] + v[k * 2 + 1]; } // 一点取得:O(log N) // i 番目の要素を返す. T get(int i) { return sum(i, i + 1); } // 区間総和:O(log N) // 半開区間 [l, r) の要素の総和を返す. T sum(int l, int r) { return sum_rf(l, r, 1, 0, n); } // sum() を実現する実際の再帰関数 // k : 注目ノード,[kl, kr) : ノード v[k] が表す区間 T sum_rf(int l, int r, int k, int kl, int kr) { // まず自身の評価を行っておく. eval(k); // 範囲外なら総和の計算に含めない. if (kr <= l || r <= kl) { return 0; } // 完全に範囲内なら葉まで降りず自身の値を返す. if (l <= kl && kr <= r) { return v[k]; } // 一部の範囲のみを含むなら子を見に行く. T sum_l = sum_rf(l, r, k * 2, kl, (kl + kr) / 2); T sum_r = sum_rf(l, r, k * 2 + 1, (kl + kr) / 2, kr); return sum_l + sum_r; } }; //【根付き木の HL 分解/オイラーツアー】O(|V|) /* * 根付き木 rt の HL 分解を行いつつオイラーツアーを得る. * * in[s] : 最重頂点優先で頂点 s に初めて入る時刻(根なら 0) * out[s] : 最重頂点優先で頂点 s から最後にでる時刻(根なら 2 |V| - 1) * pos[t] : 最重頂点優先で時刻 t で居る頂点(長さ 2 |V| - 1) * top[s] : 頂点 s を含む連結成分の最も浅い頂点 */ template <class TREE> void hld_and_et(TREE& rt, vi& in, vi& out, vi& pos, vi& top) { // 参考:https://qiita.com/Pro_ktmr/items/4e1e051ea0561772afa3 int n = (int)rt.v.size(); int time = 0; in = vi(n); out = vi(n); pos = vi(2 * n - 1); top = vi(n); // 再帰用の関数 // s : 注目している頂点 // p : s を含む連結成分の最も浅い頂点 function<void(int, int)> rf = [&](int s, int p) { in[s] = time; pos[time++] = s; top[s] = p; // 重さ最大の頂点を得る. int w_max = -INF, v_max = -1; for (auto t : rt.v[s].child) { if (chmax(w_max, rt.v[t].weight)) { v_max = t; } } // 重さ最大の頂点を優先的になぞる. if (v_max != -1) { rf(v_max, p); pos[time++] = s; } // 残りの頂点をなぞる. for (auto t : rt.v[s].child) { if (t == v_max) { continue; } rf(t, t); pos[time++] = s; } // s から最後に離れる out[s] = time; }; // 根から順に探索する. rf(rt.r, rt.r); } //【部分木加算/パス総和クエリ】 /* * subtree_add_path_sum_query(rt) : O(|V|) * 根付き木 rt で初期化する. * * add(v, val) : O(log |V|) * 頂点 v の部分木の辺に val を加算する. * * add(v1, v2, val) : O((log |V|)^2) * 頂点 v1 から v2 までの辺に val を加算する. * * sum(v1, v2) : O((log |V|)^2) * 頂点 v1 から v2 までの辺の値の和を返す. * * 利用: * 【根付き木の HL 分解/オイラーツアー】 * 【遅延評価セグメント木:区間加算/区間総和クエリ】 */ template <class TREE, class T> struct subtree_add_path_sum_query { // 参考:https://qiita.com/Pro_ktmr/items/4e1e051ea0561772afa3 // 根付き木 TREE rt; // HL 分解とオイラーツアーの結果の記録用 // in[s] : 最重頂点優先で頂点 s に初めて入る時刻(根なら 0) // out[s] : 最重頂点優先で頂点 s から最後にでる時刻(根なら 2 |V| -1) // pos[t] : 最重頂点優先で時刻 t で居る頂点(長さ 2 |V| -1) // top[s] : 頂点 s を含む連結成分の最も浅い頂点 vi in, out, pos, top; // オイラーツアーで得られた列 pos に対する区間加算/区間総和クエリを処理する. // rasq[t] : 時刻 t で居る頂点に入る辺の値(rasq[0] は使わない) range_add_sum_query<T> rasq; // コンストラクタ(根付き木で初期化) subtree_add_path_sum_query(TREE& rt_) : rt(rt_) { // rt を HL 分解しつつオイラーツアーを得る. hld_and_et(rt, in, out, pos, top); vector<T> val(2 * rt.n - 1); rep(s, rt.n) { repe(t, rt.v[s].child) { if (get_cost(t) != INFL) { val[in[t]] += t.cost; } } } rasq = range_add_sum_query<T>(val); } // 頂点 v の部分木の辺に val を加算する. void add(int v, T val) { rasq.add(in[v] + 1, out[v], val); } // 頂点 v1 から v2 までの辺に val を加算する. void add(int v1, int v2, T val) { // v1 と v2 が異なる連結成分に属している限りループを回す. while (top[v1] != top[v2]) { // v1 の方が浅い連結成分に属しているとする. if (top[v1] > top[v2]) { swap(v1, v2); } // v2 を含む連結成分は pos で並んで配置されているので, // 最も浅い頂点 top[v2] から v2 までの範囲に val を加算する. rasq.add(in[top[v2]], in[v2] + 1, val); // 一つ浅い連結成分に移動する. v2 = rt.v[top[v2]].parent; } // ここまできたら v1 と v2 は同じ連結成分に属するので, // その間の辺のみに対して val を加算する. if (v1 > v2) { swap(v1, v2); } rasq.add(in[v1] + 1, in[v2] + 1, val); } // 頂点 v1 から v2 までの辺の値の和を返す. T sum(int v1, int v2) { T res = 0; // v1 と v2 が異なる連結成分に属している限りループを回す. while (top[v1] != top[v2]) { // v1 の方が浅い連結成分に属しているとする. if (top[v1] > top[v2]) { swap(v1, v2); } // v2 を含む連結成分は pos で並んで配置されているので, // 最も浅い頂点 top[v2] から v2 までの範囲の和を求める. res += rasq.sum(in[top[v2]], in[v2] + 1); // 一つ浅い連結成分に移動する. v2 = rt.v[top[v2]].parent; } // ここまできたら v1 と v2 は同じ連結成分に属するので, // その間の辺のみの和を res に加算する. if (v1 > v2) { swap(v1, v2); } res += rasq.sum(in[v1] + 1, in[v2] + 1); return res; } }; int main() { cout << fixed << setprecision(15); // 小数点以下の桁数の指定 int n; cin >> n; WGraph g(n); read_graph(n, n - 1, g, false, false); WRTree rt(g, 0); subtree_add_path_sum_query<WRTree, ll> sapsq(rt); rep(s, n) { repe(t, rt.v[s].child) { // sapsq.add(s, t.to, t.cost); } } dump(sapsq.in); dump(sapsq.out); dump(sapsq.pos); dump(sapsq.top); dump(sapsq.rasq); int q; cin >> q; rep(i, q) { int type; cin >> type; if (type == 1) { int a; ll x; cin >> a >> x; sapsq.add(a, x); } else { int b; cin >> b; cout << sapsq.sum(0, b) << endl; } dump(sapsq.rasq); } }