二叉树的高度和深度 --代码实现

一：若只有一个根节点，看概念定义为 0层还是1层

以下代码均定义为1层即只有一个根节点时候，树的高度为1

递归实现：

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    int TreeDepth(TreeNode* pRoot)
    {
        if(pRoot==NULL)
            return 0;
        int left=TreeDepth(pRoot->left);
        int right=TreeDepth(pRoot->right);
        return (left>right?left:right)+1;
    }
};

非递归实现:

class Solution {
public:
    int TreeDepth(TreeNode* pRoot)
    {
        queue<TreeNode *>q;
        if(pRoot==NULL)
            return 0;
        q.push(pRoot);
        int depth=0;
        while(!q.empty())
        {
            int len=q.size();
            depth++;
            while(len--)
            {
                TreeNode * temp=q.front();
                q.pop();
                if(temp->left) q.push(temp->left);
                if(temp->right) q.push(temp->right);
            }
        }
        return depth;
    }
};


/* 递归函数 - 计算二叉树高度(只有一个根节点的二叉树高为1) */
 
#include <iostream>
#include <malloc.h>
using namespace std;
 
//二叉树节点定义
typedef int ElementType;
typedef struct bitnode
{
	ElementType data;                  //数据域 
	struct bitnode *left, *right;      //指针域 
} bitnode, *bitree;                    //bitnode为结构体，bitree为指向结构体bitnode的指针 
 
 
//先序创建一棵二叉树
bitree CreateTree()           
{
	bitree T;
	ElementType item;
	
	cin >> item;
	if( item == 0 )                  //0为空树标记 
	{
		T = NULL;                    //T为空树 
	}
	else
	{
		T = (bitree)malloc(sizeof(bitnode));
		T->data = item;
		
		T->left = CreateTree();              //递归创建左子树 
		T->right = CreateTree();             //递归创建右子树 
	} 
	
	return T;                              //返回根节点 
} 
 
 
//先序周游一棵树
void PerOrder(bitree T)
{
	if( T )                      //T != NULL
	{
		cout << T->data << " ";             //先访问根节点 
		PerOrder(T->left);                  //再访问其左子树 
		PerOrder(T->right);                 //最后访问其右子树 
	}
} 
 
//释放树的空间 
bitree FreeTree(bitree T)
{
	if( T )
	{
		FreeTree(T->left);                  //递归释放其左子树 
		FreeTree(T->right);                 //递归释放其右子树 
		
		free(T);                            //释放掉根节点 
		T = NULL;                           //释放掉指向根节点的指针，避免野指针 
	}
	
	return T;
}
 
//计算一棵树的高度
int TreeHeight(bitree T)
{
	if( T == NULL )           //空树，返回0 
	    return 0;
	if( T->left == NULL && T->right == NULL )          //树根返回1 
	    return 1;
	return max(TreeHeight(T->left), TreeHeight(T->right)) + 1; //树的高度 = MAX(左子树的高度，右子树的高度) + 1;
} 
 
 
//main()函数中为测试代码，读者可自行设计 
int main()
{
	bitree root;
	int height = 0;
	
	cout << "先序创建一棵二叉树，一次性输入一组数据(以空格隔开):";
	root = CreateTree();
	
	cout << "先序周游:";
	PerOrder(root);
	cout << endl;
	
	height = TreeHeight(root);
	cout << "树的高度为:" << height << endl;
	
	root = FreeTree(root);
	if(root == NULL)
	    cout << "释放成功" << endl;
	    
	return 0;
}

二叉树的高度和深度 --代码实现

猜你喜欢