洛谷P1131 [ZJOI2007] 时态同步（树形dp）

题目链接

https://www.luogu.com.cn/problem/P1131

思路

我们调整靠近根节点的树枝时，其下叶子节点距离根节点的距离都会增加。因此，调整越靠近根节点的树枝所需要付出的代价就越少。

在本题中，我们可以用节点深度来替代距离。我们可以先找到最深的叶子节点，再从最小的子树开始，通过最近的树枝，将所有的叶子节点调整到同一深度。以此类推，直到整颗树都被调整完成。

调整过程如下图所示：


每次调整的代价为：$d{p}_u-(d{p}_j+t)$
其中，$u$为当前节点，$j$为节点$u$的一个子节点，$t$为节点$u$与$j$的边的权值。

代码

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 5e5 + 5;
const int inf = 0x3f3f3f3f3f3f3f3f;
int n, s, ans;
int dp[N];
vector<pair<int, int>>mp[N];
void dfs(int u, int fu)
{
	for (int i = 0; i < mp[u].size(); i++)
	{
		int j = mp[u][i].first;
		int t = mp[u][i].second;
		if (j == fu) continue;
		dfs(j, u);
		dp[u] = max(dp[u], dp[j] + t);
	}
	for (int i = 0; i < mp[u].size(); i++)
	{
		int j = mp[u][i].first;
		int t = mp[u][i].second;
		if (j == fu) continue;
		ans += (dp[u] - (dp[j] + t));
	}
}
void solve()
{
	cin >> n >> s;
	for (int i = 1, u, v, t; i < n; i++)
	{
		cin >> u >> v >> t;
		mp[u].push_back({v, t});
		mp[v].push_back({u, t});
	}
	dfs(s, -1);
	cout << ans << endl;
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int test = 1;
	// cin >> test;
	for (int i = 1; i <= test; i++)
	{
		solve();
	}
	return 0;
}