Full Depth Morning Show(换根法， 树形dp)

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;
}