結果
問題 | No.386 貪欲な領主 |
ユーザー |
![]() |
提出日時 | 2022-03-07 20:26:58 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.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 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 16 |
ソースコード
#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;}