結果
問題 | No.386 貪欲な領主 |
ユーザー | kroton |
提出日時 | 2016-06-28 13:39:11 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 191 ms / 2,000 ms |
コード長 | 1,665 bytes |
コンパイル時間 | 828 ms |
コンパイル使用メモリ | 64,048 KB |
実行使用メモリ | 29,520 KB |
最終ジャッジ日時 | 2023-08-02 08:52:43 |
合計ジャッジ時間 | 3,070 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 6 ms
19,668 KB |
testcase_01 | AC | 5 ms
19,612 KB |
testcase_02 | AC | 5 ms
19,688 KB |
testcase_03 | AC | 5 ms
19,704 KB |
testcase_04 | AC | 191 ms
29,520 KB |
testcase_05 | AC | 132 ms
25,040 KB |
testcase_06 | AC | 140 ms
24,976 KB |
testcase_07 | AC | 6 ms
19,700 KB |
testcase_08 | AC | 22 ms
20,256 KB |
testcase_09 | AC | 7 ms
19,720 KB |
testcase_10 | AC | 5 ms
19,716 KB |
testcase_11 | AC | 5 ms
19,652 KB |
testcase_12 | AC | 6 ms
19,724 KB |
testcase_13 | AC | 8 ms
19,924 KB |
testcase_14 | AC | 140 ms
24,976 KB |
testcase_15 | AC | 162 ms
29,468 KB |
ソースコード
#include <cstdio> #include <cstring> #include <cassert> #include <vector> #include <set> #include <map> #include <queue> #include <algorithm> using namespace std; typedef long long ll; const int MAX_V = 100010; const int MAX_LOG_V = 20; int V; vector<int> G[MAX_V]; int U[MAX_V]; int parent[MAX_LOG_V][MAX_V]; int sum[MAX_LOG_V][MAX_V]; int depth[MAX_V]; void dfs(int v, int p, int d){ parent[0][v] = p; sum[0][v] = U[v]; depth[v] = d; for(int u : G[v]){ if(u != p){ dfs(u, v, d + 1); } } } void init(){ dfs(0, -1, 0); for(int k = 0; k + 1 < MAX_LOG_V; k++){ for(int v = 0; v < V; v++){ if(parent[k][v] < 0){ parent[k + 1][v] = -1; sum[k + 1][v] = sum[k][v]; } else { int p = parent[k][v]; parent[k + 1][v] = parent[k][p]; sum[k + 1][v] = sum[k][v] + sum[k][p]; } } } } int solve(int u, int v){ int res = 0; if(depth[u] > depth[v])swap(u, v); for(int k = 0; k < MAX_LOG_V; k++){ if((depth[v] - depth[u]) >> k & 1){ res += sum[k][v]; v = parent[k][v]; } } if(u == v){ return res + U[u]; } for(int k = MAX_LOG_V - 1; k >= 0; k--){ if(parent[k][u] != parent[k][v]){ res += sum[k][u] + sum[k][v]; u = parent[k][u]; v = parent[k][v]; } } int lca = parent[0][u]; res += U[u] + U[v] + U[lca]; return res; } int main(){ scanf("%d", &V); for(int i=0;i<V-1;i++){ int v, u; scanf("%d %d", &v, &u); G[v].push_back(u); G[u].push_back(v); } for(int i=0;i<V;i++){ scanf("%d", &U[i]); } init(); int q; scanf("%d", &q); ll res = 0; for(int i=0;i<q;i++){ int u, v, w; scanf("%d %d %d", &u, &v, &w); res += solve(u, v) * w; } printf("%lld\n", res); return 0; }