题解

描述

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

思路

本题的想法主要还是参照各路大神的思路。我原本的想法是遍历这棵树得到深度,但是在遍历的过程中,不知道如何实现既遍历左子树和右子树,如果遍历两次很明显代码很冗余且时间复杂度大。参照别人的思路,使用了工具--队列,存放当前结点的左右节点,同时result加1,最后当队列为空,即所有的结点都已经被遍历,得到的result就是结果。

代码

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

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

    }

}
*/
import java.util.*;
public class Solution {
    public int TreeDepth(TreeNode root) {
        int result = 0;
        if(root == null){
            return result;
        }
        
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()){
            int size = queue.size();
            while(size-->0){
                TreeNode node = queue.poll();
                if(node.left != null){
                    queue.add(node.left);
                }
                if(node.right != null){
                    queue.add(node.right);
                }
            }
            result++;
        }
        return result;
        
    }
}