剑指offer-38. 二叉树的深度

题目描述

输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。

递归解题思路:

1、如果是空树,返回0;
2、如果只有根结点,返回1;
3、否则递归遍历左右子树;
 4、返回左右子树较大者+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)
    {
        /*
        解题思路:
        1、如果是空树,返回0;
        2、如果只有根结点,返回1;
        3、否则递归遍历左右子树;
        4、返回左右子树较大者+1(根结点).
        */
        if(pRoot==NULL){
            return 0;
        }
        if(pRoot->right==NULL && pRoot->left == NULL){
            return 1;
        }
        int leftLen,rightLen;
        leftLen = TreeDepth(pRoot->left);
        rightLen = TreeDepth(pRoot->right);
        int h = leftLen>rightLen?leftLen:rightLen;
        return h+1;
    
    }
};

 效率:

非递归解题思路:

1、根结点入队列,记录当前结点数;
2、如果队列非空,队首元素出队,结点数加1;
3、左孩子非空,左孩子入队,如果右孩子非空,右孩子入队;
4、遍历完所有当前层的结点,深度加1,结点数清零,重新开始计数;
5、遍历完整棵树,队列为空,返回深度即为树的深度

代码实现:

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.LinkedList;
import java.util.Queue;
public class Solution {
    public int TreeDepth(TreeNode root) {
        /*
        解题思路:
        1、根结点入队列,记录当前结点数;
        2、如果队列非空,队首元素出队,结点数加1;
        3、左孩子非空,左孩子入队,如果右孩子非空,右孩子入队;
        4、遍历完所有当前层的结点,深度加1,结点数清零,重新开始计数;
        5、遍历完整棵树,队列为空,返回深度即为树的深度
        */
        if(root==null){
            return 0;
        }
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        int depth = 0;
        int count = 0,nextCount;
        // 1、根结点入队列,记录当前结点数;
        q.offer(root);
        nextCount = q.size();
        while(q.size()!=0){
           // 2、如果队列非空,队首元素出队,结点数加1;
           TreeNode currentNode = q.poll();
           count++;
           // 3、左孩子非空,左孩子入队,如果右孩子非空,右孩子入队;
           if(currentNode.left!=null){
                  q.offer(currentNode.left);
           }
           if(currentNode.right!=null){
                  q.offer(currentNode.right);
           }
           //4、遍历完所有当前层的结点,深度加1,结点数清零,重新开始计数
           if(count == nextCount){
               depth++;
               count = 0;
               nextCount = q.size();
           }
        }
        return depth;
    }
}

 效率:

发布了89 篇原创文章 · 获赞 0 · 访问量 937

猜你喜欢

转载自blog.csdn.net/qq_34449717/article/details/103881151