SCAU 18923 二叉树的直径

思路：显然最长路径是从一个叶子结点经过根节点走到另一个叶子结点，两个叶子分别在左右子树，如果都在一边路径会重叠不符合题意

所以问题转化为分别求出左右子树的最深深度，然后相加即可。

分别遍历左右子树即可，注意n==1特判一下，还有注意没有右儿子的情况就好了> - <

(用的链式前向星存图，看的有点懵的小伙伴可以理解为链表思想存结点即可，或者自行百度一手

#include <iostream>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0)
#define _for(i,a,b) for(int i=(a) ;i<=(b) ;i++)
#define _rep(i,a,b) for(int i=(a) ;i>=(b) ;i--)
#define mst(x,y) memset(x,y,sizeof(x))
#define all(v) v.begin() , v.end()
#define pb(v) push_back(v)
#define INF 0x3f3f3f3f
#define int long long
#define lson p*2,l,mid
#define rson p*2+1,mid+1,r
typedef long long ll;
const int N = 1e5+10;
int n;
int head[N];
int maxn=1;//左子树最大深度，除了n==1，根节点必有左儿子
int maxn2;//右子树最大深度
vector <int > dp[100];//每层结点数
struct ty
{
    int t,next;
}edge[N];
int tot;
void addedge(int x ,int y)
{
    edge[++tot].t = y;
    edge[tot].next = head[x];
    head[x] = tot;
}
//遍历左子树
void dfs1( int x ,int fa, int p)//当前结点编号 ， 当前结点父亲，当前深度
{
    for(int i=head[x] ;i!=0 ;i=edge[i].next)
    {
        int y = edge[i].t;
        if( y==fa) continue;//如果访问过了，跳过
        maxn = max(maxn, p+1);//维护最深叶子结点
        dfs1(y,x,p+1);
    }

}
//遍历右子树
void dfs2( int x ,int fa, int p)//当前结点编号 ， 当前结点父亲，当前深度
{
    int cnt=0;
    for(int i=head[x] ;i!=0 ;i=edge[i].next)
    {
        int y = edge[i].t;
        if( y==fa) continue;//如果访问过了，跳过
        maxn2 = max(maxn2, p+1);//维护最深叶子结点
        dfs2(y,x,p+1);

    }

}

signed main()
{
    //!!!!!!!!!!!!!!!!!!!!!!
//    freopen("data.txt","r",stdin);
    //!!!!!!!!!!!!!!!!!!!!!!
    IOS;
    cin>>n;
    //先用链式前向星存树
    _for(i,1,n-1)
    {
        int x, y;
        cin>>x>>y;
        addedge(x,y);
        addedge(y,x);
    }
    //一个结点特判
    if(n==1)
    {
        cout<<0<<endl;
        return 0;
    }
    int left = edge[head[1]].t;//根节点左儿子
    int right =-1;
    if( edge[edge[head[1]].next].t!=0)//如果根节点有右儿子
    {
        right = edge[edge[head[1]].next].t;
    }
    //如果有根节点右儿子需要交换预处理一下，由于先来的点存在后面，
    //此时left其实是右儿子，right是左儿子
    if( right!=-1)
    {
        swap(left,right);
    }
    //遍历左子树
    dfs1(left,1,1);
    //有右儿子的话遍历右子树
    if( right!=-1 )
    {
        maxn2=1;//避免右子树就一个根节点
        dfs2(right,1,1);
    }
    dfs2(right,1,1);
    cout<<maxn+maxn2<<endl;
}