結果

問題 No.386 貪欲な領主
ユーザー Manuel1024Manuel1024
提出日時 2022-03-07 20:26:58
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 309 ms / 2,000 ms
コード長 2,336 bytes
コンパイル時間 894 ms
コンパイル使用メモリ 84,140 KB
実行使用メモリ 16,768 KB
最終ジャッジ日時 2024-07-22 15:48:28
合計ジャッジ時間 4,073 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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 309 ms
16,576 KB
testcase_05 AC 238 ms
16,768 KB
testcase_06 AC 246 ms
16,756 KB
testcase_07 AC 4 ms
5,376 KB
testcase_08 AC 35 ms
5,376 KB
testcase_09 AC 5 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 4 ms
5,376 KB
testcase_13 AC 8 ms
5,376 KB
testcase_14 AC 253 ms
16,708 KB
testcase_15 AC 266 ms
16,576 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
using ll = long long;

struct LCA{
    vector<vector<int>> parent;
    vector<int> dist;

    LCA(const vector<vector<int>> &G, int root = 0){
        init(G, root);
    }

    void init(const vector<vector<int>> &G, int root = 0){
        dist.assign(G.size(), -1);
        int k = 1;
        for(; (1 << k) < G.size(); k++);
        parent.assign(k, vector<int>(G.size(), -1));

        queue<int> Q;
        dist[root] = 0;
        Q.emplace(root);
        while(!Q.empty()){
            int c = Q.front(); Q.pop();
            for(const auto &nex: G[c]){
                if(dist[nex] != -1) continue;
                dist[nex] = dist[c]+1;
                parent[0][nex] = c;
                Q.emplace(nex);
            }
        }
        for(int i = 1; i < parent.size(); i++){
            for(int v = 0; v < G.size(); v++){
                if(parent[i-1][v] < 0) parent[i][v] = 0;
                else parent[i][v] = parent[i-1][parent[i-1][v]];
            }
        }
    }

    int query(int u, int v){
        if(dist[u] < dist[v]) swap(u, v);
        for(int i = 0; i < parent.size(); i++){
            if((dist[u]-dist[v]) >> i & 1){
                u = parent[i][u];
            }
        }

        if(u == v) return u;
        for(int i = parent.size()-1; i >= 0; i--){
            if(parent[i][u] != parent[i][v]){
                u = parent[i][u];
                v = parent[i][v];
            }
        }
        return parent[0][u];
    }
};

int main(){
    int n; cin >> n;
    vector<vector<int>> G(n);
    for(int i = 0; i < n-1; i++){
        int a, b; cin >> a >> b;
        G[a].emplace_back(b);
        G[b].emplace_back(a);
    }
    vector<int> u(n);
    for(auto &it: u) cin >> it;

    LCA lca(G);
    vector<int> cost(n, -1);
    cost[0] = u[0];
    queue<int> Q;
    Q.emplace(0);
    while(!Q.empty()){
        int c = Q.front(); Q.pop();
        for(auto const &nex: G[c]){
            if(cost[nex] != -1) continue;
            cost[nex] = cost[c]+u[nex];
            Q.emplace(nex);
        }
    }

    int m; cin >> m;
    ll ans = 0;
    while(m--){
        int a, b, c; cin >> a >> b >> c;
        ans += (ll)(cost[a]+cost[b]-2*cost[lca.query(a, b)]+u[lca.query(a, b)])*c;
    }
    cout << ans << endl;
    return 0;
}
0