leetcode-102 二叉树的层次遍历 JAVA代码

给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

[
  [3],
  [9,20],
  [15,7]
]

JAVA代码

    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        List<Integer> temList = new ArrayList<Integer>();
        if (root == null) {
            return result;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        ((LinkedList<TreeNode>) queue).add(root);
        int num = 1;
        int next = 0;
        while (!queue.isEmpty()) {
            TreeNode now = queue.poll();
            temList.add(now.val);
            num--;
            if (now.left != null) {
                ((LinkedList<TreeNode>) queue).add(now.left);
                next++;
            }
            if (now.right != null) {
                ((LinkedList<TreeNode>) queue).add(now.right);
                next++;
            }
            if (num == 0) {
                num = next;
                next = 0;
                result.add(temList);
                temList = new ArrayList<Integer>();
            }
        }
        return result;
    }

利用队列先进先出的一个特性进行一次二叉树的层序遍历

发布了31 篇原创文章 · 获赞 12 · 访问量 6549

猜你喜欢

转载自blog.csdn.net/cyz52/article/details/97557511