求一棵树两个点的最远距离

最新推荐文章于 2024-08-09 20:38:37 发布

JK Chen

最新推荐文章于 2024-08-09 20:38:37 发布

阅读量1.6k

点赞数 1

分类专栏：图论/搜索

本文链接：https://blog.csdn.net/jk_chen_acmer/article/details/79480370

版权

图论/搜索专栏收录该内容

113 篇文章 4 订阅

订阅专栏

原题：1361 树的最远距离

题意：

给一棵树，求两个点这间的距离的最大值

解析：

想了很久，发现两个点a，b也就只有几种情况

a为b的父结点
a与b属于同一父结点的两条不同分支

看到题目，很多人当然会想到离root最远的那个结点。我们现在就利用这个结点来设计算法。

证明：最远路一点以最远结点为端点

如果是第一种情况，a，b为同一分支，那么a和b同时往相反方向走到极限，便是一个为root结点（此题为1），一个为末结点，那么这个末结点一定是最远结点。

第二种情况的话，利用反证法，分两种讨论

要选两个结点非最远结点且与最远结点同父结点
如果不选3而选4和5，可以看出，2通往3代替2通往5才是最优解
要选两个结点非最远结点且与最远结点非同一父结点
选5和6时，改选3的话不止加上第一种情况的2~5，更是多了一段2~4，更应该选5

所以我们定下了一个点，最远结点。

另外一个点就是到最远结点的最远的点，把最远结点看成root结点再找一遍即可

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
#include<list>
#include<vector>
#include<stack>
#include<queue>
#include<ctime>
#include<cstdlib>
#include<sstream>
//#include<windows.h> 
#include<functional>
#define D long long
#define F double
#define MAX 0x7fffffff
#define MIN -0x7fffffff
#define mmm(a,b) memset(a,b,sizeof(a))
#define pb push_back  
#define mk make_pair  
#define fi first  
#define se second  
#define pill pair<int, int>  
#define for1(i,a,b) for(int i=a;i<=b;i++)
#define for2(i,a,b) for(int i=a;i>=b;i--)
#define ini(n) scanf("%d",&n)
#define inll(n) scanf("%lld",&n)
#define outisp(n) printf("%d ",n)
#define outllsp(n) printf("%lld ",n)
#define outiel(n) printf("%d\n",n)
#define outllel(n) printf("%lld\n",n)
using namespace std; 
#define N 60100 
#define MOD ((int)1e9+7)
#define random(a,b) (rand()%(b-a+1)+a)
#define stop Sleep(2000)
#define CLS system("cls")
const string el="\n";
const string elel="\n\n";
const string sp=" ";
const string spsp="  ";
const string tab="\t";
const F eps=1e-6;

int n;
int pos1,pos2;

int head[100009];
int now;
struct edge{
    int to,nex,val;
}e[200009];
void add(int a,int b,int v){
    e[++now].to=b;e[now].val=v;e[now].nex=head[a];head[a]=now;
    e[++now].to=a;e[now].val=v;e[now].nex=head[b];head[b]=now;
}

int vis[100009];
struct node{
    int _id,v;node(int _id,int v):_id(_id),v(v){}
};
int maxlen;
int finmax(int root){
    maxlen=0;int maxpos;
    mmm(vis,0);
    queue<node>Q;
    Q.push(node(root,0));
    while(!Q.empty()){
        node t=Q.front();Q.pop();
        if(vis[t._id])continue;vis[t._id]=1;
        if(t.v>maxlen)maxlen=t.v,maxpos=t._id;
        for(int i=head[t._id];~i;i=e[i].nex){
            Q.push(node(e[i].to,t.v+e[i].val));
        }
    }
    return maxpos;
}

int main(){
    mmm(head,-1);ini(n);for1(i,1,n-1){
        int a,b,v;scanf("%d%d%d",&a,&b,&v);add(a,b,v);
    }
    int root=finmax(1);
    finmax(root);
    outiel(maxlen);
}

/*
6
1 2 2
2 3 5
2 4 1
4 5 3
4 6 2

*/