XTU OJ Highway(树的直径)

树的直径是指树中两个节点间的最长路径。通过两次深度优先搜索(DFS)可以找到它:首先从任意节点出发找到最长路径的终点,然后从该终点再次DFS,第二次DFS的最长路径即为树的直径。这种方法基于两点:1) 如果初始节点在直径上,第二次DFS的终点是直径的另一端;2) 如果初始节点不在直径上,从初始节点到终点的路径会与直径相交,使得终点仍是直径的一部分。最终,求解所有节点到两个直径端点距离的最大值之和减去直径长度。

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

树的直径是指树的最长简单路。
求法: 两遍DFS :先任选一个起点DFS找到最长路的终点,再从终点进行DFS,则第二次DFS找到的最长路即为树的直径;
原理: 设起点为u,第一次DFS找到的终点v一定是树的直径的一个端点
证明:
1) 如果u 是直径上的点,则v显然是直径的终点(因为如果v不是的话,则必定存在另一个点w使得u到w的距离更长,则于DFS找到了v矛盾)
2) 如果u不是直径上的点,则u到v必然于树的直径相交(反证),那么交点到v 必然就是直径的后半段了,所以v一定是直径的一个端点,所以从v进行DFS得到的一定是直径长度

这题的解法就是通过第一次DFS找到树的直径的一个端点Start,然后从Start开始用一次DFS找到直径上的另一个端点End(算出每个点到Start的距离),最后再从End开始DFS一次(算出每个点到End的距离),最后求解的就是所有点到直径端点的最大距离的和再减去一个直径的距离(因为直径累加了两次,所以要减去一次)

#include <map>
#include <set>
#include <cmath>
#include <ctime>
#include <stack>
#include <queue>
#include <cstdio>
#include <memory>
#include <cctype>
#include <bitset>
#include <string>
#include <vector>
#include <climits>
#include <cstring>
#include <iostream>
#include <iomanip>
#include <algorithm>
#include <functional>

//#define FIN freopen("input.txt","r",stdin);
//#define FOUT freopen("output.txt","w+",stdout);
using namespace std;
typedef long long ll;

const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const double eps=1e-8;
const double Pi=acos(-1.0);
const int N=100100;

struct Edge
{
    int to,w;
};
vector<Edge> G[N];
bool vis[N];
int n,Start,End;
ll dis1[N],dis2[N],d[N],maxcost;
void dfs(int u)
{
    vis[u] = 1;
    int size = G[u].size();     //与顶点u相连的点数
    for (int i = 0; i<size; i++)             //对与顶点u相连的点数进行扫描
    {
        int v = G[u][i].to;
        if (!vis[v])
        {
            d[v] = d[u] + G[u][i].w;
            dfs(v);
        }
    }
}
void init()
{
        memset(vis,0,sizeof(vis));
        memset(d,0,sizeof(d));
}
int main()
{
    while(~scanf("%d",&n))
    {
        memset(dis1,0,sizeof(dis1));
        memset(dis2,0,sizeof(dis2));
        for(int i=0; i<N; i++)
            G[i].clear();
        int from,to,cost;
        for(int i=0; i<n-1; i++)
        {
            scanf("%d %d %d",&from,&to,&cost);
            G[from].push_back((Edge){to,cost});
            G[to].push_back((Edge){from,cost});
        }
        init();
        dfs(1);
        maxcost=-1;
        for(int i=1;i<=n;i++)
        {
            if(d[i]!=0&&maxcost<d[i])
            {
                maxcost=d[i];
                Start=i;
            }
        }
        init();
        dfs(Start);
        maxcost=-1;
        for(int i=1;i<=n;i++)
        {
            dis1[i]=d[i];
            if(d[i]!=0&&maxcost<d[i])
            {
                End=i;
                maxcost=d[i];
            }
        }
        init();
        dfs(End);
        memcpy(dis2,d,sizeof(d));
//        for(int i=1;i<=n;i++)
//            printf("%d ",dis1[i]);
//        cout<<endl;
//        for(int i=1;i<=n;i++)
//            printf("%d ",dis2[i]);
//        cout<<endl;
//        cout<<maxcost<<endl;
        ll ans=0;
        for(int i=1;i<=n;i++)
            ans+=max(dis1[i],dis2[i]);
        printf("%I64d\n",ans-maxcost);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值