站外题,代码求调悬1关
  • 板块学术版
  • 楼主lskksl
  • 当前回复8
  • 已保存回复8
  • 发布时间2023/7/5 20:25
  • 上次更新2023/11/3 11:26:06
查看原帖
站外题,代码求调悬1关
260074
lskksl楼主2023/7/5 20:25

题目描述 给定一颗树和根节点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;
}

2023/7/5 20:25
加载中...