一棵树,给一个数组num[1...n],如果i是根,i拥有num[i]个“孩子”,否则i拥有num[i]-1个“孩子”。问在所有树的可能形态中,可能的最远的两个点的距离。首先应该想到,num[i]实际上表示的是点i有多少条边,就能直接判断出树是否合法了。然后答案就是n-数组中1的个数+1。因为度大于1的那些点,可以连成一串,在这一串的头尾补上度为1的点就行了,这肯定是合法的最长距离。
题虽然不涉及什么算法,但是还是要经过思考才能得到解法,比较有意思。
#include <bits/stdc++.h>
using namespace std;
#define ll long long
class VampireTree{
public:
int maxDistance(vector <int> num){
int sz = num.size();
int sum = 0;
for(int i=0;i<sz;i++){
sum+=num[i];
}
if(sum!=(sz-1)*2)return -1;
int count_one=0;
for(int i=0;i<sz;i++){
if(num[i] == 1)count_one++;
}
return sz-count_one+1;
}
};