結果
問題 | No.399 動的な領主 |
ユーザー | BinomialSheep |
提出日時 | 2023-03-21 23:44:11 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 425 ms / 2,000 ms |
コード長 | 11,518 bytes |
コンパイル時間 | 2,496 ms |
コンパイル使用メモリ | 216,720 KB |
実行使用メモリ | 23,576 KB |
最終ジャッジ日時 | 2024-09-18 14:34:24 |
合計ジャッジ時間 | 7,025 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
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 | 4 ms
5,376 KB |
testcase_05 | AC | 25 ms
5,376 KB |
testcase_06 | AC | 425 ms
17,408 KB |
testcase_07 | AC | 415 ms
17,368 KB |
testcase_08 | AC | 412 ms
17,536 KB |
testcase_09 | AC | 424 ms
17,480 KB |
testcase_10 | AC | 4 ms
5,376 KB |
testcase_11 | AC | 21 ms
5,376 KB |
testcase_12 | AC | 310 ms
17,872 KB |
testcase_13 | AC | 301 ms
17,916 KB |
testcase_14 | AC | 89 ms
23,552 KB |
testcase_15 | AC | 112 ms
23,576 KB |
testcase_16 | AC | 168 ms
20,416 KB |
testcase_17 | AC | 417 ms
17,408 KB |
testcase_18 | AC | 412 ms
17,408 KB |
ソースコード
#pragma region header #include <bits/stdc++.h> // デバッグ用マクロ:https://naskya.net/post/0002/ #ifdef LOCAL #include <debug_print.hpp> #define debug(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__) #else #define debug(...) (static_cast<void>(0)) #endif using namespace std; using ll = long long; using vi = vector<int>; using vl = vector<long long>; using vs = vector<string>; using vc = vector<char>; using vb = vector<bool>; using vpii = vector<pair<int, int>>; using vpll = vector<pair<long long, long long>>; using vvi = vector<vector<int>>; using vvl = vector<vector<long long>>; using vvc = vector<vector<char>>; using vvb = vector<vector<bool>>; using vvvi = vector<vector<vector<int>>>; using pii = pair<int, int>; #define rep(i, n) for (int i = 0; i < (int)(n); i++) #define all(x) (x).begin(), (x).end() #define INT(...) \ int __VA_ARGS__; \ in(__VA_ARGS__) #define LL(...) \ ll __VA_ARGS__; \ in(__VA_ARGS__) #define STR(...) \ string __VA_ARGS__; \ in(__VA_ARGS__) #define Sort(a) sort(all(a)) #define VEC(type, name, size) \ vector<type> name(size); \ in(name) [[maybe_unused]] void print() {} template <class T, class... Ts> void print(const T& t, const Ts&... ts); template <class... Ts> void out(const Ts&... ts) { print(ts...); cout << '\n'; } namespace IO { #define VOID(a) decltype(void(a)) struct S { S() { cin.tie(nullptr)->sync_with_stdio(0); fixed(cout).precision(12); } } S; template <int I> struct P : P<I - 1> {}; template <> struct P<0> {}; template <class T> void i(T& t) { i(t, P<3>{}); } void i(vector<bool>::reference t, P<3>) { int a; i(a); t = a; } template <class T> auto i(T& t, P<2>) -> VOID(cin >> t) { cin >> t; } template <class T> auto i(T& t, P<1>) -> VOID(begin(t)) { for (auto&& x : t) i(x); } template <class T, size_t... idx> void ituple(T& t, index_sequence<idx...>) { in(get<idx>(t)...); } template <class T> auto i(T& t, P<0>) -> VOID(tuple_size<T>{}) { ituple(t, make_index_sequence<tuple_size<T>::value>{}); } template <class T> void o(const T& t) { o(t, P<4>{}); } template <size_t N> void o(const char (&t)[N], P<4>) { cout << t; } template <class T, size_t N> void o(const T (&t)[N], P<3>) { o(t[0]); for (size_t i = 1; i < N; i++) { o(' '); o(t[i]); } } template <class T> auto o(const T& t, P<2>) -> VOID(cout << t) { cout << t; } template <class T> auto o(const T& t, P<1>) -> VOID(begin(t)) { bool first = 1; for (auto&& x : t) { if (first) first = 0; else o(' '); o(x); } } template <class T, size_t... idx> void otuple(const T& t, index_sequence<idx...>) { print(get<idx>(t)...); } template <class T> auto o(T& t, P<0>) -> VOID(tuple_size<T>{}) { otuple(t, make_index_sequence<tuple_size<T>::value>{}); } #undef VOID } // namespace IO #define unpack(a) \ (void)initializer_list<int> { (a, 0)... } template <class... Ts> void in(Ts&... t) { unpack(IO::i(t)); } template <class T, class... Ts> void print(const T& t, const Ts&... ts) { IO::o(t); unpack(IO::o((cout << ' ', ts))); } #undef unpack // #define MAX 10000 #define INFTY (1 << 30) // 浮動小数点の誤差を考慮した等式 #define EPS (1e-10) #define equal(a, b) (fabs((a) - (b)) < EPS) template <typename T> inline bool chmax(T& a, T b) { return ((a < b) ? (a = b, true) : (false)); } template <typename T> inline bool chmin(T& a, T b) { return ((a > b) ? (a = b, true) : (false)); } #define YESNO(yes, no) \ void yes(bool i = 1) { out(i ? #yes : #no); } \ void no() { out(#no); } YESNO(first, second) YESNO(First, Second) YESNO(Yes, No) YESNO(YES, NO) YESNO(possible, impossible) YESNO(POSSIBLE, IMPOSSIBLE) #pragma endregion header // #include <atcoder/all> // using namespace atcoder; #pragma region acl_internal_bit namespace internal { // @param n `0 <= n` // @return minimum non-negative `x` s.t. `n <= 2**x` int ceil_pow2(int n) { int x = 0; while ((1U << x) < (unsigned int)(n)) x++; return x; } // @param n `1 <= n` // @return minimum non-negative `x` s.t. `(n & (1 << x)) != 0` int bsf(unsigned int n) { #ifdef _MSC_VER unsigned long index; _BitScanForward(&index, n); return index; #else return __builtin_ctz(n); #endif } } // namespace internal #pragma endregion acl_internal_bit #pragma region acl_lazySeg namespace myAtcoder { template <class S, S (*op)(S, S), S (*e)(), class F, S (*mapping)(F, S), F (*composition)(F, F), F (*id)()> struct lazy_segtree { public: lazy_segtree() : lazy_segtree(0) {} lazy_segtree(int n) : lazy_segtree(std::vector<S>(n, e())) {} lazy_segtree(const std::vector<S>& v) : _n(int(v.size())) { log = internal::ceil_pow2(_n); size = 1 << log; d = std::vector<S>(2 * size, e()); lz = std::vector<F>(size, id()); for (int i = 0; i < _n; i++) d[size + i] = v[i]; for (int i = size - 1; i >= 1; i--) { update(i); } } void set(int p, S x) { assert(0 <= p && p < _n); p += size; for (int i = log; i >= 1; i--) push(p >> i); d[p] = x; for (int i = 1; i <= log; i++) update(p >> i); } S get(int p) { assert(0 <= p && p < _n); p += size; for (int i = log; i >= 1; i--) push(p >> i); return d[p]; } S prod(int l, int r) { assert(0 <= l && l <= r && r <= _n); if (l == r) return e(); l += size; r += size; for (int i = log; i >= 1; i--) { if (((l >> i) << i) != l) push(l >> i); if (((r >> i) << i) != r) push(r >> i); } S sml = e(), smr = e(); while (l < r) { if (l & 1) sml = op(sml, d[l++]); if (r & 1) smr = op(d[--r], smr); l >>= 1; r >>= 1; } return op(sml, smr); } S all_prod() { return d[1]; } void apply(int p, F f) { assert(0 <= p && p < _n); p += size; for (int i = log; i >= 1; i--) push(p >> i); d[p] = mapping(f, d[p]); for (int i = 1; i <= log; i++) update(p >> i); } void apply(int l, int r, F f) { assert(0 <= l && l <= r && r <= _n); if (l == r) return; l += size; r += size; for (int i = log; i >= 1; i--) { if (((l >> i) << i) != l) push(l >> i); if (((r >> i) << i) != r) push((r - 1) >> i); } { int l2 = l, r2 = r; while (l < r) { if (l & 1) all_apply(l++, f); if (r & 1) all_apply(--r, f); l >>= 1; r >>= 1; } l = l2; r = r2; } for (int i = 1; i <= log; i++) { if (((l >> i) << i) != l) update(l >> i); if (((r >> i) << i) != r) update((r - 1) >> i); } } template <bool (*g)(S)> int max_right(int l) { return max_right(l, [](S x) { return g(x); }); } template <class G> int max_right(int l, G g) { assert(0 <= l && l <= _n); assert(g(e())); if (l == _n) return _n; l += size; for (int i = log; i >= 1; i--) push(l >> i); S sm = e(); do { while (l % 2 == 0) l >>= 1; if (!g(op(sm, d[l]))) { while (l < size) { push(l); l = (2 * l); if (g(op(sm, d[l]))) { sm = op(sm, d[l]); l++; } } return l - size; } sm = op(sm, d[l]); l++; } while ((l & -l) != l); return _n; } template <bool (*g)(S)> int min_left(int r) { return min_left(r, [](S x) { return g(x); }); } template <class G> int min_left(int r, G g) { assert(0 <= r && r <= _n); assert(g(e())); if (r == 0) return 0; r += size; for (int i = log; i >= 1; i--) push((r - 1) >> i); S sm = e(); do { r--; while (r > 1 && (r % 2)) r >>= 1; if (!g(op(d[r], sm))) { while (r < size) { push(r); r = (2 * r + 1); if (g(op(d[r], sm))) { sm = op(d[r], sm); r--; } } return r + 1 - size; } sm = op(d[r], sm); } while ((r & -r) != r); return 0; } private: int _n, size, log; std::vector<S> d; std::vector<F> lz; void update(int k) { d[k] = op(d[2 * k], d[2 * k + 1]); } void all_apply(int k, F f) { d[k] = mapping(f, d[k]); if (k < size) lz[k] = composition(f, lz[k]); } void push(int k) { all_apply(2 * k, lz[k]); all_apply(2 * k + 1, lz[k]); lz[k] = id(); } }; } // namespace myAtcoder #pragma endregion acl_lazySeg // HL分解(Heavy Light Decomposition) // beet-aizuさんから窃盗 // https://beet-aizu.github.io/library/tree/heavylightdecomposition.cpp // BEGIN CUT HERE class HLD { private: void dfs_sz(int v) { auto& es = G[v]; if (~par[v]) es.erase(find(es.begin(), es.end(), par[v])); for (int& u : es) { par[u] = v; dfs_sz(u); sub[v] += sub[u]; if (sub[u] > sub[es[0]]) swap(u, es[0]); } } void dfs_hld(int v, int& pos) { vid[v] = pos++; inv[vid[v]] = v; for (int u : G[v]) { if (u == par[v]) continue; nxt[u] = (u == G[v][0] ? nxt[v] : u); dfs_hld(u, pos); } } public: vector<vector<int>> G; // vid: vertex -> idx // inv: idx -> vertex vector<int> vid, nxt, sub, par, inv; HLD(int n) : G(n), vid(n, -1), nxt(n), sub(n, 1), par(n, -1), inv(n) {} void add_edge(int u, int v) { G[u].emplace_back(v); G[v].emplace_back(u); } void build(int r = 0) { int pos = 0; dfs_sz(r); nxt[r] = r; dfs_hld(r, pos); } int lca(int u, int v) { while (1) { if (vid[u] > vid[v]) swap(u, v); if (nxt[u] == nxt[v]) return u; v = par[nxt[v]]; } } template <typename F> void for_each(int u, int v, const F& f) { while (1) { if (vid[u] > vid[v]) swap(u, v); f(max(vid[nxt[v]], vid[u]), vid[v] + 1); if (nxt[u] != nxt[v]) v = par[nxt[v]]; else break; } } template <typename F> void for_each_edge(int u, int v, const F& f) { while (1) { if (vid[u] > vid[v]) swap(u, v); if (nxt[u] != nxt[v]) { f(vid[nxt[v]], vid[v] + 1); v = par[nxt[v]]; } else { if (u != v) f(vid[u] + 1, vid[v] + 1); break; } } } }; // END CUT HERE /* No.399 動的な領主 HLD + 遅延セグ木で殴ります いもす法で解けるが、それはそれで大変 実はyukicoderはACLが使えるのでACLから抜き出す必要はないらしい */ // struct S { // ll value; // int size; // }; // using F = ll; // S e() { return {0, 0}; } // S mapping(F f, S x) { return {x.value + f * x.size, x.size}; } // F composition(F f, F g) { return f + g; } // F id() { return 0; } struct S { long long value; int size; }; using F = long long; S op(S a, S b) { return {a.value + b.value, a.size + b.size}; } S e() { return {0, 0}; } S mapping(F f, S x) { return {x.value + f * x.size, x.size}; } F composition(F f, F g) { return f + g; } F id() { return 0; } int main() { /* input */ INT(N); HLD G(N); rep(i, N - 1) { INT(u, v); --u, --v; G.add_edge(u, v); } /* preprocessing */ G.build(); // 遅延セグ木を宣言(区間加算、区間和取得) // vector<S> v(N, {0, 1}); std::vector<S> v(N, {0, 1}); myAtcoder::lazy_segtree<S, op, e, F, mapping, composition, id> seg(v); /* query */ INT(Q); while (Q--) { INT(A, B); --A, --B; G.for_each(A, B, [&](int l, int r) -> void { seg.apply(l, r, 1); }); } /* output */ ll ans = 0; auto f = [](ll x) { return x * (1 + x) / 2; }; rep(i, N) ans += f(seg.get(i).value); out(ans); return 0; }