hdu 4118 Holiday's Accommodation 树形dp

来源：互联网发布：长城证券软件下载编辑：程序博客网时间：2024/06/11 22:19

点击打开链接

题意：

给出一个无向图,现在把所有的点的人都交换,保证每个地方只有一个人,且任何人都不在自己原来的那个点上,问交换的过程中所有人走的最远的距离是多少;

思路：

首先分析一下，我们对每一个边进行分析，每个边的左边有n个节点，右边有m个节点，那么必然ans+=min(n,m)*边权

仔细想想，就很清楚，假设左边节点比右边少，那么我让左边的节点都到右边去，一定最优。

树形dp：dp[u]表示包括u节点的子节点个数，对于红色的那条边 dp[2]=6, dp[4]=4。取min(4, n-4)*w*2

代码：

#include <bits/stdc++.h>using namespace std;typedef long long ll;const int maxn = 1e5+10;vector<pair<int,int> > G[maxn];ll dp[maxn],ans;int a[maxn];struct edge{int x,y,z;}E[maxn];void dfs(int u){dp[u] += 1;for(int i=0; i<G[u].size(); i++){int v = G[u][i].first;if(dp[v] == 0){dfs(v);dp[u] += dp[v];}}}int main(){int T; cin>>T;for(int cas=1; cas<=T; cas++){memset(dp,0,sizeof(dp));ans = 0;int n; cin>>n;for(int i=0; i<=n; i++) G[i].clear();for(int i=1; i<n; i++){cin >> E[i].x >> E[i].y >> E[i].z;G[E[i].x].push_back(make_pair(E[i].y,E[i].z));G[E[i].y].push_back(make_pair(E[i].x,E[i].z));}dfs(1);for(int i=1; i<n; i++){int k = min(dp[E[i].x],dp[E[i].y]);ans += min(k,n-k) * E[i].z * 2;}cout << "Case #" << cas << ": " << ans << endl;}}

0 0