题目描述 给定一颗树和根节点S,记dist(S,i)为叶子节点i到S的最短路径;我们将该路径上的所有的边权w乘起来作为i到S的代价;请你求出最大代价和最小代价所对应的树的路径;若有多个代价最大的叶子节点,取编号最大的叶子编号,记为A,若有多个代价最小的叶子节点,取编号最小的叶子节点编号,记为B,请输出A和B以及dist(A,B)中,深度最浅的节点编号。
输入
5 1
1 2 3
1 3 7
2 4 3
2 5 1
输出
4 5 2
数据范围:1<=n<=1e5,1<=z<=1e4(z为边权)
#include<bits/stdc++.h>
using namespace std;
vector< pair<int,int> >vec[100010];
int vis[100010];
long long ans=1,maxx=0,minn=2147483648;
int a,b=100001,c;
void dfs(int x)
{
if(vec[x].size()==1)
{
if(maxx<=ans)
{
if(maxx==ans) a=max(a,x);
else a=x;
maxx=ans;
}
if(minn>=ans)
{
if(minn==ans) b=min(b,x);
else b=x;
minn=ans;
}
return;
}
vis[x]=1;
for(int i=0;i<vec[x].size();i++)
{
int y=vec[x][i].first;
int z=vec[x][i].second;
if(vis[y]==0)
{
ans*=z;
dfs(y);
ans/=z;
}
}
vis[x]=0;
}
void dfs2(int x,int now)
{
if(x==b)
{
c=now;
return;
}
vis[x]=1;
for(int i=0;i<vec[x].size();i++)
{
int y=vec[x][i].first;
if(vis[y]==0)
{
if(now<x) dfs2(y,now);
else dfs2(y,x);
}
}
vis[x]=0;
}
int main()
{
//freopen(".in","r",stdin);
//freopen(".out","w",stdout);
int n,s;
scanf("%d%d",&n,&s);
for(int i=1;i<=n-1;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
vec[x].push_back(make_pair(y,z));
vec[y].push_back(make_pair(x,z));
}
dfs(s);
dfs2(a,a);
printf("%d %d %d\n",a,b,c);
//fclose(stdin);
//fclose(stdout);
return 0;
}