結果

問題 No.399 動的な領主
コンテスト
ユーザー zaki_joho
提出日時 2019-10-22 22:14:45
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 702 ms / 2,000 ms
コード長 6,530 bytes
コンパイル時間 2,131 ms
コンパイル使用メモリ 188,248 KB
実行使用メモリ 21,632 KB
最終ジャッジ日時 2024-07-05 00:00:47
合計ジャッジ時間 9,066 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"

class HLDecomposition
{
    int n, pos;
    std::vector<std::vector<int>> g;
    // vid:=HLD後のグラフでのid, head:=属するheavy path の head, sub:=子孫数
    // hvy:=heavy path 上の次の頂点, par:=parent, dep:=depth, inv:=vidの逆, type:=森用
    std::vector<int> vid, head, sub, hvy, par, dep, inv, type;

    void dfs(int cur)
    {
        for (auto nxt : g[cur])
        {
            if (nxt == par[cur])
                continue;
            par[nxt] = cur;
            dep[nxt] = dep[cur] + 1;
            dfs(nxt);
        }
        int ret = 0;
        for (auto u : g[cur])
        {
            if (u == par[cur])
                continue;
            sub[cur] += sub[u];
            if (ret < sub[u])
            {
                ret = sub[u];
                hvy[cur] = u;
            }
        }
    }

    void bfs(int r, int c)
    {
        std::queue<int> que({r});
        while (!que.empty())
        {
            int h = que.front();
            que.pop();
            for (int i = h; i != -1; i = hvy[i])
            {
                type[i] = c;
                vid[i] = pos++;
                inv[vid[i]] = i;
                head[i] = h;
                for (int j : g[i])
                    if (j != par[i] && j != hvy[i])
                        que.push(j);
            }
        }
    }

public:
    HLDecomposition() {}
    HLDecomposition(int sz) : n(sz), pos(0), g(n),
                              vid(n, -1), head(n), sub(n, 1), hvy(n, -1),
                              par(n, -1), dep(n, 0), inv(n), type(n) {}

    void add_edge(int u, int v)
    {
        g[u].push_back(v);
        g[v].push_back(u);
    }

    void build(std::vector<int> rs = std::vector<int>(1, 0))
    {
        int c = 0;
        for (int r : rs)
        {
            dfs(r);
            bfs(r, c++);
        }
    }

    // f(l, r) := [l, r] に対する演算
    void for_each_vertex(int u, int v, const std::function<void(int, int)> &f)
    {
        while (true)
        {
            if (vid[u] > vid[v])
                std::swap(u, v);
            f(std::max(vid[head[v]], vid[u]), vid[v]);
            if (head[u] == head[v])
                break;
            v = par[head[v]];
        }
    }

    // f(l, r) := [l, r] に対する演算
    void for_each_edge(int u, int v, const std::function<void(int, int)> &f)
    {
        while (true)
        {
            if (vid[u] > vid[v])
                std::swap(u, v);
            if (head[u] == head[v])
            {
                if (u != v)
                    f(vid[u] + 1, vid[v]);
                break;
            }
            f(vid[head[v]], vid[v]);
            v = par[head[v]];
        }
    }

    int lca(int u, int v)
    {
        while (true)
        {
            if (vid[u] > vid[v])
                std::swap(u, v);
            if (head[u] == head[v])
                return u;
            v = par[head[v]];
        }
    }

    int distance(int u, int v)
    {
        return dep[u] + dep[v] - 2 * dep[lca(u, v)];
    }
};

// range add range sum
class LazySegmentTree
{
    int n;
    using T1 = long long;
    using T2 = long long;
    std::vector<T1> node;
    std::vector<T2> lazy;

    // 遅延評価
    void eval(int k, int l, int r)
    {
        // 遅延配列が空なら終了
        if (lazy[k] == 0)
            return;
        // 遅延配列を適用
        node[k] = node[k] + lazy[k];
        if (r - l > 1)
        {
            lazy[2 * k + 1] = lazy[2 * k + 1] + lazy[k] / 2;
            lazy[2 * k + 2] = lazy[2 * k + 2] + lazy[k] / 2;
        }
        // 遅延配列初期化
        lazy[k] = 0;
    }

public:
    LazySegmentTree(int _n)
    {
        int sz = _n;
        n = 1;
        while (n < sz)
            n *= 2;
        // 配列初期化
        node.resize(2 * n - 1, 0);
        lazy.resize(2 * n - 1, 0);
    }
    LazySegmentTree(int _n, T1 _v)
    {
        int sz = _n;
        n = 1;
        while (n < sz)
            n *= 2;
        node.resize(2 * n - 1, 0);
        lazy.resize(2 * n - 1, 0);

        for (int i = 0; i < sz; i++)
            node[i + n - 1] = _v;
        for (int i = n - 2; i >= 0; i--)
            node[i] = node[i * 2 + 1] + node[i * 2 + 2];
    }

    // 半開区間 [a, b) に対して値 val を反映させる
    void update(int a, int b, T2 val, int l = 0, int r = -1, int k = 0)
    {
        if (r < 0)
            r = n;
        // ノード k で遅延評価
        eval(k, l, r);
        if (b <= l || r <= a)
            return;
        // 区間が被覆されている場合
        if (a <= l && r <= b)
        {
            // 遅延配列更新, 評価
            lazy[k] = lazy[k] + (r - l) * val;
            eval(k, l, r);
        }
        else
        {
            // 子ノードの値を評価し, 更新
            int mid = (l + r) / 2;
            update(a, b, val, l, mid, 2 * k + 1);
            update(a, b, val, mid, r, 2 * k + 2);
            node[k] = node[2 * k + 1] + node[2 * k + 2];
        }
    }

    // 半開区間 [a, b) に対してクエリを投げる
    T1 query(int a, int b, int l = 0, int r = -1, int k = 0)
    {
        if (r < 0)
            r = n;
        eval(k, l, r);
        // 範囲外なら単位元返す
        if (b <= l || r <= a)
            return 0;
        if (a <= l && r <= b)
            return node[k];
        int mid = (l + r) / 2;
        T1 vl = query(a, b, l, mid, 2 * k + 1);
        T1 vr = query(a, b, mid, r, 2 * k + 2);
        return vl + vr;
    }
};

using namespace std;

// 頂点クエリテストしてない
void solve_yukicoder_399()
{
    using ll = long long;
    int n;
    cin >> n;
    HLDecomposition hld(n);
    LazySegmentTree lst(n);
    for (int i = 0; i < n - 1; i++)
    {
        int u, v;
        cin >> u >> v;
        u--;
        v--;
        // cerr << u << " " << v << endl;
        hld.add_edge(u, v);
    }
    hld.build();
    for (int i = 0; i < n; i++)
    {
        hld.for_each_vertex(i, i, [&](int l, int r) {
            lst.update(l, r + 1, 1);
        });
    }
    int q;
    cin >> q;
    ll ret = 0;
    while (q--)
    {
        int a, b;
        cin >> a >> b;
        a--;
        b--;
        hld.for_each_vertex(a, b, [&](int l, int r) {
            ret += lst.query(l, r + 1);
        });
        hld.for_each_vertex(a, b, [&](int l, int r) {
            lst.update(l, r + 1, 1);
        });
    }
    cout << ret << endl;
}

int main()
{
    solve_yukicoder_399();
}
0