Link
题目大意: 给一颗树, 树中每个点都有一个权值, 每个边都有一个边权,定义两个点的距离为d(u, v) = (u和v的路径上边权之和乘以u和v的点权之和)现求每个点到其他所有点的距离。
思路:由于数据范围过大, 显然不能暴力枚举两个点之间的距离, 而且如果暴力枚举中显然有大量的数据被重复计算, 显然联想到树形dp, 用换根法进行转移。
细节见于代码,一种传参, 一种记录值
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e5 + 10;
int h[N], e[N *2], f[N * 2], ne[N * 2];
int n;
int idx;
int w[N], num[N], sum[N], ans[N], dis[N], num2[N];
void add(int u, int v, int x)
{
e[idx] = v, ne[idx] = h[u], f[idx] = x, h[u] = idx++;
e[idx] = u, ne[idx] = h[v], f[idx] = x, h[v] = idx++;
}
void dfs(int u, int fa)
{
num[u] = w[u];
num2[u] = 1;
for(int i = h[u]; ~i; i = ne[i])
{
int t = e[i];
if(t == fa) continue;
dfs(t, u);
num2[u] += num2[t];// num2 点数
dis[u] += f[i] * num2[t] + dis[t];
sum[u] += sum[t] + f[i] * num[t];
num[u] += num[t];
}
}
int p_num, s_num;
void dp(int u, int fa, int fa_sum, int fa_dis)
{
int a[5];
for(int i = h[u]; ~i; i = ne[i])
{
int t = e[i];
if(t == fa) continue;
int tmp_dis = dis[t] + (fa_dis - (dis[t] + num2[t] * f[ i])) + (n - num2[t]) * f[i];
int tmp_sum = sum[t] + (fa_sum - (sum[t] + num[t]*f[i])) + (s_num - num[t])*f[i];
ans[t] = w[t] * tmp_dis + tmp_sum;
dp(t, u, tmp_sum, tmp_dis);
}
}
signed main()
{
scanf("%lld", &n);
memset(h, -1, sizeof h);
for(int i = 1; i <= n; i++) scanf("%lld", w + i), s_num += w[i];
for(int i = 1; i < n; i++)
{
int x, y, z;
scanf("%lld%lld%lld", &x, &y, &z);
add(x, y, z);
}
dfs(1, 0);
ans[1] = w[1]*dis[1] + sum[1];
p_num = n; //总点数
dp(1, 0, sum[1], dis[1]);
for(int i = 1; i <= n; i++)
{
printf("%lld\n", ans[i]);
}
return 0;
}
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e5 + 10;
int h[N], e[N *2], f[N * 2], ne[N * 2];
int n;
int idx;
int w[N], num[N], sum[N], ans[N], dis[N], num2[N];
void add(int u, int v, int x)
{
e[idx] = v, ne[idx] = h[u], f[idx] = x, h[u] = idx++;
e[idx] = u, ne[idx] = h[v], f[idx] = x, h[v] = idx++;
}
void dfs(int u, int fa)
{
num[u] = w[u];
num2[u] = 1;
for(int i = h[u]; ~i; i = ne[i])
{
int t = e[i];
if(t == fa) continue;
dfs(t, u);
num2[u] += num2[t];// num2 点数
dis[u] += f[i] * num2[t] + dis[t];
sum[u] += sum[t] + f[i] * num[t];
num[u] += num[t];
}
}
int p_num, s_num;
void dp(int u, int fa, int fa_sum, int fa_dis)
{
int a[5];
for(int i = h[u]; ~i; i = ne[i])
{
int t = e[i];
if(t == fa) continue;
int tmp_dis = dis[t] + (fa_dis - (dis[t] + num2[t] * f[ i])) + (n - num2[t]) * f[i];
int tmp_sum = sum[t] + (fa_sum - (sum[t] + num[t]*f[i])) + (s_num - num[t])*f[i];
ans[t] = w[t] * tmp_dis + tmp_sum;
dp(t, u, tmp_sum, tmp_dis);
}
}
signed main()
{
scanf("%lld", &n);
memset(h, -1, sizeof h);
for(int i = 1; i <= n; i++) scanf("%lld", w + i), s_num += w[i];
for(int i = 1; i < n; i++)
{
int x, y, z;
scanf("%lld%lld%lld", &x, &y, &z);
add(x, y, z);
}
dfs(1, 0);
ans[1] = w[1]*dis[1] + sum[1];
p_num = n; //总点数
dp(1, 0, sum[1], dis[1]);
for(int i = 1; i <= n; i++)
{
printf("%lld\n", ans[i]);
}
return 0;
}