树的相关操作

最新推荐文章于 2022-01-18 08:59:00 发布

yellowjianokok

最新推荐文章于 2022-01-18 08:59:00 发布

阅读量251

点赞数

分类专栏：树

本文链接：https://blog.csdn.net/yellowjianokok/article/details/48865861

版权

ACM 同时被 2 个专栏收录

40 篇文章 0 订阅

订阅专栏

树

2 篇文章 0 订阅

订阅专栏

树的直径对于树上的边带有权重的话而且每个节点最多只能经过一次的话我们定义树的直径是俩个点最大距离为树的直径然后其他的点的最大距离为到俩个直径端点的最大的那个

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<vector>
using namespace std;
typedef long long ll;
typedef pair<int,int> pp;
#define inf 0x3f3f3f3f
#define eps 1e-10
#define maxl 30010
#define mem(i,j) memset(i,j,sizeof(i))
const int mod=1e9+7;
struct edge{
    int to,cost;
    edge(int to,int cost):to(to),cost(cost){}
    edge(){}
};
vector<edge> g[maxl];
int d[maxl],vis[maxl];
int res,ans;
int n,m;//节点和边的数目
void bfs(int p){
    queue<int> que;
    memset(d,0,sizeof(d));
    memset(vis,0,sizeof(vis));
    while(que.size()) que.pop();
    res=p;ans=0;
    que.push(p);
    vis[p]=1;
    while(que.size()){
        int q=que.front();que.pop();
        for(int i=0;i<g[q].size();i++){
            edge tmp=g[q][i];
            if(!vis[tmp.to]){
                d[tmp.to]=d[q]+tmp.cost;
                vis[tmp.to]=1;
                que.push(tmp.to);
                if(ans<d[tmp.to]){
                    ans=d[tmp.to];
                    res=tmp.to;
                }
            }
        }
    }
}
int main()
{
    freopen("in.txt", "r", stdin);
    scanf("%d%d",&n,&m);
    int u,v,w;
    for(int i=1;i<=n;i++) g[i].clear();
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&u,&v,&w);
        g[u].push_back(edge(v,w));
        g[v].push_back(edge(u,w));
    }
    bfs(1);
    bfs(res);
    printf("%d\n",ans);
}

题目： hdu 4607