tree.cpp

二叉树

// Tree.cpp: implementation of the CTree class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "Tree.h"

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CTree::CTree()
{
	m_pRoot=NULL;
	m_TreeNodeNum=0;
}

CTree::~CTree()
{
	m_TreeNodeNum=0;
	m_pRoot=NULL;
}

long CTree::GetTreeNodeNum() const
{
	return m_TreeNodeNum;
}

void CTree::CheckNodeNum(CTreeNode *pRoot)
{
	if(pRoot)
	{
		m_TreeNodeNum++;

		CheckNodeNum(pRoot->m_pLeft);
		CheckNodeNum(pRoot->m_pRight);
	}
}

long CTree::CheckDepth(CTreeNode *pRoot)
{
	if((pRoot->m_pLeft==NULL)&&(pRoot->m_pRight==NULL))
	{
		return 1;
	}
	else
	{
		long depth1=0;
		if(pRoot->m_pLeft)
			depth1=CheckDepth(pRoot->m_pLeft)+1;

		long depth2=0;
		if(pRoot->m_pRight)
			depth2=CheckDepth(pRoot->m_pRight)+1;

		return (depth1>depth2?depth1:depth2);
	}
}