比赛 |
CSP2022提高组 |
评测结果 |
WWWWWWWWWWWTTWWWTTTWWWTTT |
题目名称 |
数据传输 |
最终得分 |
0 |
用户昵称 |
该账号已注销 |
运行时间 |
25.827 s |
代码语言 |
C++ |
内存使用 |
0.00 MiB |
提交时间 |
2022-10-30 11:04:46 |
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;
long long n, q, k, v[200100], cnt = 0, hd[200100];
long long d[200100], f[200100];
struct edge {
int t, n;
} e[400100];
void add(int x, int y) {
e[++cnt].t = y;
e[cnt].n = hd[x];
hd[x] = cnt;
}
int dfs(int x, int fa) {
d[x] = d[fa] + 1;
f[x] = fa;
for (int i = hd[x]; i; i = e[i].n) {
int ver = e[i].t;
if (ver == fa)
continue;
dfs(ver, x);
}
return 0;
}
int lca(int x, int y) {
int s = x, e = y;
long long ans = 0;
if (d[x] != d[y]) {
while (d[x] < d[y]) {
ans += v[y];
y = f[y];
}
}
if (x == y)
return ans * 2 + v[y] * 2 - v[s] - v[e];
while (x != y) {
ans += v[x];
ans += v[y];
x = f[x];
y = f[y];
}
return ans * 2 + 2 * v[x] - v[s] - v[e];
}
int main() {
freopen("csp2022_transmit.in", "r", stdin);
freopen("csp2022_transmit.out", "w", stdout);
cin >> n >> q >> k;
for (int i = 1; i <= n; i++) {
cin >> v[i];
}
for (int i = 1; i < n; i++) {
int x, y;
cin >> x >> y;
add(x, y);
add(y, x);
}
d[1] = 1;
dfs(1, 1);
for (int i = 1; i <= q; i++) {
int x, y;
cin >> x >> y;
cout << lca(x, y) << endl;
}
return 0;
}