洛谷P2052 [NOI2011] 道路修建题解

原创于 2026-03-23 10:08:11 发布 · 732 阅读

18 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

收录于

洛谷题解

1.原题和原题链接

原题链接

P2052 [NOI2011] 道路修建

题目描述

在 W 星球上有 $n$ 个国家。为了各自国家的经济发展，他们决定在各个国家之间建设双向道路使得国家之间连通。但是每个国家的国王都很吝啬，他们只愿意修建恰好 $n - 1$ 条双向道路。

每条道路的修建都要付出一定的费用，这个费用等于道路长度乘以道路两端的国家个数之差的绝对值。例如，在下图中，虚线所示道路两端分别有 $2$ 个、 $4$ 个国家，如果该道路长度为 $1$ ，则费用为 $1 \times ∣2 - 4∣ = 2$ 。图中圆圈里的数字表示国家的编号。

由于国家的数量十分庞大，道路的建造方案有很多种，同时每种方案的修建费用难以用人工计算，国王们决定找人设计一个软件，对于给定的建造方案，计算出所需要的费用。请你帮助国王们设计一个这样的软件。

输入格式

输入的第一行包含一个整数 $n$ ，表示 W 星球上的国家的数量，国家从 $1$ 到 $n$ 编号。

接下来 $n - 1$ 行描述道路建设情况，其中第 $i$ 行包含三个整数 $a_i$ ， $b_i$ 和 $c_i$ ，表示第 $i$ 条双向道路修建在 $a_i$ 与 $b_i$ 两个国家之间，长度为 $c_i$ 。

输出格式

输出一个整数，表示修建所有道路所需要的总费用。

输入输出样例 #1

输入 #1

输出 #1

说明/提示

对于 $100\%$ 的数据， $1\leq a_i, b_i\leq n$ ， $0\leq c_i\leq10^6$ ， $2\leq n\leq 10^6$ 。

测试点编号	$n =$
$1$	$2$
$2$	$10$
$3$	$100$
$4$	$200$
$5$	$500$
$6$	$600$
$7$	$800$
$8$	$1000$
$9$	$10^4$
$10$	$2\times 10^4$
$11$	$5\times 10^4$
$12$	$6\times 10^4$
$13$	$8\times 10^4$
$14$	$10^5$
$15$	$6\times 10^5$
$16$	$7\times 10^5$
$17$	$8\times 10^5$
$18$	$9\times 10^5$
$19, 20$	$10^6$

2.主要思路

先创建一个动态数组g，类型为pair，方便捆绑存储国家之间的连接和道路长度。一个数组dp记录一个国家所在的一边共有多少个国家。一个变量s记录耗费费用。并使用深搜传入2个数据，即当前国家和上一个国家，先将dp数组当前位置设为1，即这个国家，再遍历所有与这个国家相连的国家，并存入变量v，w存入与这个国家的距离，再判断是否为上一个国家，是则跳过本次循环，再深搜传入v，k，并将结果dp[v]加到dp[k]中，即可求出此国家这边的国家总数，而另一边的国家数量就是国家总数减去此国家这边的国家数量，故将s加上w*abs(n-dp[v]*2)，最后出深搜就可以输出最终的结果s了。

3.AC代码

#include<bits/stdc++.h>
using namespace std;
long long n,x,y,z,s=0,dp[1000005];
vector<pair<int,int>> g[1000005];
void dfs(int k,int fa){
	dp[k]=1;
	for (int i=0;i<g[k].size();i++){
		long long v=g[k][i].first;
		long long w=g[k][i].second;
		if (v==fa) continue;
		dfs(v,k);
		dp[k]+=dp[v];
		s+=w*abs(n-dp[v]*2);
	}
}
int main(){
    cin>>n;
    for (int i=0;i<n-1;i++){
    	cin>>x>>y>>z;
    	g[x].push_back({y,z});
    	g[y].push_back({x,z});	
	}
	dfs(1,0);cout<<s;
    return 0;
}