51Nod 2602 树的直径

Problem

一棵树的直径就是这棵树上存在的最长路径。现在有一棵n个节点的树，现在想知道这棵树的直径包含的边的个数是多少？

Solution

随便找一个点，找最远的，再找新的点最远的。

当然我瞎搞的树状dp，子树分支最大的和次大的加起来就行。

Code

#include<stdio.h>
#include<set>
#include<iostream>
#include<stack>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
typedef long long ll;
typedef long double ld;
typedef double db;
#define io_opt ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
using namespace std;
const int mod=998244353;
int mo(ll a,int p){
    return a>=p?a%p:a;
}
inline int rd() {
    int x = 0, f = 1;
    char ch;
    while (ch < '0' || ch > '9') {
        if (ch == '-')f = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9') {
        x = x * 10 + ch - '0';
        ch = getchar();
    }
    return f * x;
}
int n,ans;
struct E{
    int u,v,nex;
}e[200020];
int g[100020];
int f[100020];
int dfs(int x){
    int mx1=0,mx2=0;
    for(int i=g[x];i>0;i=e[i].nex){
        if(!f[e[i].v]){
            f[e[i].v]=1;
            int d=dfs(e[i].v);
            f[e[i].v]=0;
            if(!mx1){
                mx1=d;
            }
            else if(!mx2){
                mx2=d;
                if(mx2>mx1) swap(mx1,mx2);
            }
            else{
                if(d>mx2) mx2=d;
                if(mx2>mx1) swap(mx1,mx2);
            }
        }
    }
    ans=max(ans,mx1+mx2+1);
    if(!mx1&&!mx2){
        return 1;
    }
    return mx1+1;
}
int main(){
    //io_opt;
    n=rd();
    int x,y;
    for(int i=1;i<=n-1;i++){
        x=rd(),y=rd();
        e[i]=(E){x,y,g[x]};g[x]=i;
        e[i+n-1]=(E){y,x,g[y]};g[y]=i+n-1;
    }
    f[1]=1;
    dfs(1);
    printf("%d\n",ans-1);
    return 0;
}