题目描述
在 W 星球上有 n 个国家。为了各自国家的经济发展,他们决定在各个国家 之间建设双向道路使得国家之间连通。但是每个国家的国王都很吝啬,他们只愿 意修建恰好 n – 1 条双向道路。 每条道路的修建都要付出一定的费用,这个费用等于道路长度乘以道路两端 的国家个数之差的绝对值。例如,在下图中,虚线所示道路两端分别有 2 个、4 个国家,如果该道路长度为 1,则费用为 1×|2 – 4|=2。图中圆圈里的数字表示国 家的编号。
由于国家的数量十分庞大,道路的建造方案有很多种,同时每种方案的修建 费用难以用人工计算,国王们决定找人设计一个软件,对于给定的建造方案,计 算出所需要的费用。请你帮助国王们设计一个这样的软件。
输入输出格式
输入格式
输入的第一行包含一个整数 n,表示 W 星球上的国家的数量,国家从 1 到 n 编号。 接下来 n – 1 行描述道路建设情况,其中第 i 行包含三个整数 ai、bi和 ci,表 示第 i 条双向道路修建在 ai与 bi两个国家之间,长度为 ci。
输出格式
输出一个整数,表示修建所有道路所需要的总费用。
输入输出样例
输入样例 #1
6
1 2 1
1 3 1
1 4 2
6 3 1
5 2 1
输出样例 #1
20
说明
1≤ai, bi≤n
0≤ci≤10^6
2≤n≤10^6
思路
对于这一道题而言,我们可以提取出来该题考查的重要的一点就是求出每一个点的子树的大小。
那么子树的大小就可以想到是:dfs从下向上遍历,统计每个子树的大小。但是,显然,这一道题并没有那么简单,这道题把求子树的大小变了一下型:求的是用一条边把n个点分成两部分的差值。
在2019/06/27的时候,我第一次提交这道题,最开始我的思路就是做n遍dfs,把每一个点都当做根节点遍历一遍,用一个二维数组保存两点之间的差值,这样子的话,是可以过掉八个点,也就是40分。
这一道题磨了我大概一天的时间,在2019/06/27的时候,我想出了另一种思路:
- 以一号节点作为根节点做一遍dfs,这样子的话就求出了在一号节点下的每个节点的子树大小。
那么我要做的就是如何把这些数值变成是两个节点之间的差值。
再之后我想到的思路是:
- 首先找出两个节点(i,j)之间的子树最小的那一个点(i),那么此时size[i]就是i的子树大小。(此时j的子树大小并不是size[j],因为size[j]的数值是等于size[i]+1,而并不是(i,j)这一条边的非i点的另一部分。)
- 其次因为一条边把n个点分成了两个部分,所以点的总数是n,那么size[j]实际上=n-size[i]。
这样子的话,两个点的差值=size[j]-size[i]。
最后再O(n)扫一遍每一条边,累计答案即可。
(当然,今天我直接想到的思路也是第二个思路)
错因
之前的错因不是很记得,所以记录一下今天错的原因
- 两点之间的差值在相减的时候没有加上绝对值,从而出现有负数。
在初始状态下的size[i]=6,size[j]=8,n=10,这样子的话,我求出的n-size[i]-size[i]=-2,实际上应该等于+2。
code
#include<bits/stdc++.h>
using namespace std;
const int nn=1e6+3;
typedef long long ll;
int n;
ll sum=0;
int size[nn];
bool vis[nn];
struct edge
{
int fr,to,ver,nex;
}e[2*nn];
int num=0;
int last[nn];
void add(int x,int y,int z)
{
e[++num].fr=x; e[num].to=y; e[num].ver=z;
e[num].nex=last[x]; last[x]=num;
}
inline int read()
{
int x=0,f=1; char ch=getchar();
while(ch>'9'||ch<'0'){if(ch=='-') f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0'; ch=getchar();}
return x*f;
}
void dfs(int x)
{
size[x]=1;
for(int i=last[x];i;i=e[i].nex)
{
int y=e[i].to;
if(vis[y]) continue;
vis[y]=true;
dfs(y);
size[x]+=size[y];
}
}
int main()
{
n=read();
for(int i=1;i<n;++i)
{
int a=read(), b=read(), c=read();
add(a,b,c); add(b,a,c);
}
vis[1]=true;
dfs(1);
for(int i=1;i<=num;i+=2)
{
int x=e[i].fr, y=e[i].to, z=e[i].ver;
if(size[x]<size[y]) swap(x,y);/*始终保证x的子树大小大于y的*/
ll ans=abs(n-size[y]-size[y]);
ans*=z;
sum+=ans;
}
printf("%lld\n",sum);
return 0;
}
在写这道题的过程中,石神说还有换根DP,但是,我不会……以后有机会会学一下,然后更换根DP。
“无论我们以后生疏成什么样子,请记得曾经我对你的好是真的。剩下的路就不陪你走了,要照顾好自己,不是每个人都是我,三生有幸,遇见你,纵使悲凉也是情。”
——出自网易云音乐《越来越不懂》热评
完……