TopCoder SRM674 div1 250

        一棵树,给一个数组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;
	}
};



你可能感兴趣的:(TopCoder SRM674 div1 250)