UVALive-6056/Gym-101472I Tree 树的直径

最新推荐文章于 2025-08-07 20:58:17 发布

原创最新推荐文章于 2025-08-07 20:58:17 发布 · 290 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ACM #算法

STL、数据结构专栏收录该内容

11 篇文章

订阅专栏

博客围绕一道题展开，利用树的直径及相关结论解题，提到据大佬说法跑三遍dfs即可，但作者仍有疑惑。此外，相同代码在gym通过，在uva上却未通过，作者希望有大佬能解答代码WA的原因。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这道题主要利用了树的直径和一条结论：距某个点最远的叶子节点一定是树的某一条直径的端点。

所以据某大佬说，跑三遍dfs就行了emmmmm..但是我到现在还是有点懵。。

另外很迷的一点是这份代码在gym上过了，但是在uva上WA了。。一模一样的代码啊...真是个玄学问题。。放在这里如果有大佬知道wa在什么地方了，还请不吝赐教（*＾-＾*）

代码如下：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<string>
#include<sstream>
#include<fstream>
#include<utility>
using namespace std;
#define ll long long
typedef pair<int,int>pp;
#define mkp make_pair
#define pb push_back
const double pi=acos(-1.0);
const double eps=1e-9;
const int INF=0x3f3f3f3f;
const ll MOD=1e9+7ll;

const int MAX=50005;
int n;
int st,dis;
int ans[MAX],tmp[MAX];
bool vis[MAX];
vector<pp>vt[MAX];

void dfs(int u,int dist)
{
    tmp[u]=dist;
    if(dist>dis)
    {
        dis=dist;
        st=u;
    }
    for(int i=0;i<vt[u].size();i++)
    {
        int v=vt[u][i].first;
        if(!vis[v])
        {
            vis[v]=true;
            dfs(v,vt[u][i].second+dist);
        }
    }
}

int main()
{
    //freopen("I.IN","r",stdin);
    while(scanf("%d",&n)==1)
    {
        int u,v,w;
        for(int i=0;i<n-1;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            u--,v--;
            vt[u].pb(mkp(v,w));
            vt[v].pb(mkp(u,w));
        }
        memset(ans,0,sizeof(ans));
        st=0,dis=0;
        memset(vis,false,sizeof(vis));
        vis[st]=true;
        dfs(st,0);
        for(int i=0;i<n;i++)
            ans[i]=max(ans[i],tmp[i]);

        dis=0;
        memset(vis,false,sizeof(vis));
        vis[st]=true;
        dfs(st,0);
        for(int i=0;i<n;i++)
            ans[i]=max(ans[i],tmp[i]);

        dis=0;
        memset(vis,false,sizeof(vis));
        vis[st]=true;
        dfs(st,0);
        for(int i=0;i<n;i++)
        {
            printf("%d\n",max(ans[i],tmp[i]));
        }
    }
	return 0;
}