【树直径】 POJ 1985 Cow Marathon

Whyckck

于 2018-09-10 10:33:57 发布

阅读量282

点赞数

CC 4.0 BY-SA版权

分类专栏：树直径

本文链接：https://blog.youkuaiyun.com/Whyckck/article/details/82586180

树直径专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种求解树直径的经典算法实现，通过两次DFS遍历找到树中最长路径。使用了C++语言，并且利用vector结合pair来存储图数据结构。文章提供了完整的代码示例，适用于竞赛编程和图论学习。

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

树直径模板

用到了vectpr中捆绑pair : https://blog.youkuaiyun.com/computer055maxi/article/details/6129736

#include <iostream>
#include <algorithm>
#include <queue>
#include <cstring>
#pragma GCC optimize(2)
using namespace std;
#define maxn 50005
#define inf 1e18
//#define mod 1e9+7
typedef long long ll;
const ll mod = 1e9+7;
ll n,m,ans,flag[maxn],point;
vector < pair<ll,ll> > arr[maxn];
void slove(ll now,ll step)
{
    if(flag[now] == 1)
        return ;
    else
        flag[now] = 1;

    if(step > ans)
    {
        ans = step;
        point = now;
    }


    for(ll i = 0; i < arr[now].size(); i++)
    {
        slove(arr[now][i].first,arr[now][i].second+step);
    }

    return ;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);

    cin >> n >> m;

    for(ll i = 0; i < m; i++)
    {
        ll a,b,c;
        char s;
        cin >> a >> b >> c >> s;
        arr[a].push_back( make_pair(b,c)  );
        arr[b].push_back( make_pair(a,c)  );
    }

    slove(1,0);

    ans = 0;
    memset(flag,0,sizeof(flag));

    slove(point,0);

    cout << ans << endl;

    return 0;
}