相关题目:树的最长路径
分析
根据题意,本题需要我们求一个树的最长路径d(树的直径),并计算总花费s。
花费计算公式:
树形Dp求树的直径的方法:
-
任取一点作为起始点k,找到距离该点最远的一个点v。
-
从点v开始搜,找到距离点v最远的一点u,则uv间的距离是树的直径。
注意最后答案会很大,要用long long。
C++ 代码
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
typedef long long LL;
int n,a,b,c;
int h[N],e[2*N],w[2*N],ne[2*N],idx;
LL ans;
void add(int a,int b,int c) //链表存储边
{
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
LL dfs(int u,int father) //搜索距离该点最远的距离
{
LL dist=0;
LL d1=0,d2=0;
for(int i=h[u];i!=-1;i=ne[i])
{
int j=e[i];
if(j!=father) //当前点不是其父亲,才继续判断,否则会产生回路
{
LL d=dfs(j,u)+w[i];
dist=max(dist,d);
if(d>d1) d2=d1,d1=d;
else if(d>d2) d2=d;
}
}
ans=max(ans,d1+d2);
return dist; //返回到该点距离最远的距离
}
int main()
{
memset(h,-1,sizeof h);
ios::sync_with_stdio(false);
cin>>n;
for(int i=0;i<n;i++)
{
cin>>a>>b>>c;
add(a,b,c),add(b,a,c);
}
dfs(1,-1); //任取一点开始搜,这里默认从1号点搜,其没有父结点,父亲为-1
ans=(21+ans)*ans/2; //计算总花费
cout<<ans;
return 0;
}
好像过不了
这个d1、d2是什么意思呀
最长和次长
谢谢~我现在已经理解了~
其实只有n-1一个输入 不过不影响ac
考试可以用到❤️
🉑以的