CF1453E Dog Snacks(贪心)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
typedef pair<int,int> plll;
const int N=4e5+10;
const int inf=0x3f3f3f3f;
int n,h[N],ne[N],e[N],idx;
int res;
int f[N];
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u,int fa){
    int i;
    int mx=0;
    int cnt=0;
    int mx1=0;
    for(i=h[u];i!=-1;i=ne[i]){
        int j=e[i];
        if(j==fa)
            continue;
        dfs(j,u);
        cnt++;
        f[u]=min(f[u],f[j]+1);
        if(f[j]+1>mx){
            mx1=mx;
            mx=f[j]+1;
        }
        else if(f[j]+1>mx1){
            mx1=f[j]+1;
        }
    }
    if(cnt==0){
        f[u]=0;
        return ;
    }
    if(u==1){
        if(cnt==1){
            res=max(res,mx);
        }
        else{
            res=max(res,mx);
            res=max(res,mx1+1);
        }
    }
    else{
        if(cnt>=2)
        res=max(res,mx+1);
    }
}
int main(){
    ios::sync_with_stdio(false);
    int t;
    cin>>t;
    while(t--){
        cin>>n;
        int i;
        res=0;
        for(i=1;i<=n;i++){
            h[i]=-1;
            f[i]=0x3f3f3f3f;
        }
        for(i=1;i<n;i++){
            int a,b;
            cin>>a>>b;
            add(a,b);
            add(b,a);
        }
        dfs(1,-1);
        cout<<res<<endl;
    }
    return 0;
}