这道题进一步理解题意是与最大的点相邻的点+1,其余所有点+2。
那么解只可能是MAXN,MAXN+1,MAXN+2中的一个。
假设MAXN只有一个,那么如果存在MAXN-1,都与MAXN连通,那么结果就是MAXN,如果存在MAXN-1不与MAXN连通,那么最终答案是MAXN+1。
假设MAXN有多个,如果所有MAXN,都连在同一个点上,那么答案是MAXN+1,否则,为MAXN+2。
这道题的核心是把无根树转为有根树,然后用dfs遍历树,使得时间复杂度为O(n)。主要是存储每个节点的儿子。下面附上代码
vector<int> G[maxn];
void read_tree(){
int u,v;
scanf("%d",&n);
for(int i = 0;i<n-1;i++){
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
}
然后就是遍历代码。
void dfs(int u,int fa){
int d = G[u].size();
for(int i = 0;i<d;i++){
int v = G[u][i];
if(v!=fa) dfs(v,p[v] = u);
}//其中fa是父节点。
}
然后就是如果根据题意,在搜索过程中找到值为MAXN及MAXN-1的个数,准确的说是每个点儿子是MAXN及MAXN-1的个数。考虑到如果有多个MAXN的情况,那么还应该保存点父亲的MAXN及MAXN-1的个数,当然还有它自己。
void dfs(int u,int fa){
v[u] = 1;
for(int i=0;i<G[u].size();++i){
int num = G[u][i];
if(v[u]) continue;
if(w[num]==MAXN) n1[u]++;
if(w[num]==MAXN-1) n2[u]++;
dfs(G[u][i],u);
}
if(w[fa]==maxw)++n1[u];
if(w[fa]==maxw-1)++n2[u];
if(w[u]==maxw)++n1[u];
if(w[u]==maxw-1)++n2[u];
}
搜索部分完成。
然后是判断。
先计算所有节点中MAXN,MAXN-1的个数。
MAXN = 1 -> MAXN-1都与MAXN相邻,那么为 MAXN,否则为MAXN+1
MAXN>1 -> 如果都连在同一个点 -> MAXN+1 否则为MAXN+2
for(int i=1;i<=n;++i){
if(w[i]==maxw&&cnt1==1&&num2[i]==cnt2){
ans = maxw;
break;
}
if(num1[i]==cnt1){
ans=maxw+1;
}
}