构建二叉树并查二叉树深度

 

输入：连接的数量n；

          连接节点：（1-2，2-3，1-3）；

输出：连接节点的数量（3）和二叉树的深度（2）。

#include <IOSTREAM>
#include <VECTOR>
#define max(a,b) a>b?a:b
using namespace std;

int main()
{
	vector<int> a,b;
	int n,cnt=0;
	cin>>n;
	vector<int> f(n+1);//用来存储数的结构，f[n]表示n的父节点
	a.push_back(0);
	b.push_back(0);
	for (int i=0;i<n;i++)
	{
		int a1,b1;
		cin>>a1>>b1;
		a.push_back(a1);b.push_back(b1);
		
	}
	for ( i=0;i<=n;i++)
		f[i]=i;
	for ( i=1;i<=n;i++)
	{
        
		if (f[a[i]]!=f[b[i]])//判断父节点以检验连通性
		{
			cnt++;
			f[b[i]]=a[i];//将b[i]的父节点置为a[i]
		}
		
	}
	vector<int> d(n+1,0);//当前节点的实时深度
	int maxlen=0;//二叉树深度
	for ( i=0;i<n;i++)//回溯求深度
	{
		d[i+1]=d[f[i]]+1;
		maxlen=max(maxlen,d[i+1]);
	}
	cout<<cnt<<endl<<maxlen<<endl;
	return 0;
}

在判断连通性那块，还可以计算连接节点所需的最大代价。