计蒜客之道2019初赛第一场第一题: 商汤的AI伴游小精灵

题目链接

此题题面简单，数据量要求不大，可通过用C++的vector暴力求解，基本思想就是树形结构求两个边数最多的节点。

注意两个节点可考虑先后次序之分，考虑先后次序则求第二个节点之前需要先对节点边数进行处理（毕竟剪掉第一个节点后可能出现某些子节点无父节点从而少一条边的情况）。

若不考虑先后次序则需要自行判断几种特殊情况。

这里代码是考虑节点先后次序的情况。

#include"bits/stdc++.h"
using namespace std;

int main(){
    int n;
    cin>>n;
    vector<int> v[n+1];
    int cnt[n+1];
    memset(cnt,0,sizeof(int)*(n+1));
    int x,y;
    int ans=0;
    for(int i=1;i<n;++i){
        cin>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
        cnt[x]++;
        cnt[y]++;
    }
    int max1=0;
    for(int i=1;i<=n;++i){
        if(cnt[max1]<cnt[i]){
            max1=i;
        }
    }
    ans+=cnt[max1];
    cnt[max1]=0;
    for(int i=0;i<v[max1].size();++i){
        cnt[v[max1][i]]--;
    }
    max1=0;
    for(int i=1;i<=n;++i){
        if(cnt[max1]<cnt[i]){
            max1=i;
        }
    }
    ans+=cnt[max1];
    ans--;
    cout<<ans;
    return 0;
}