LeetCode-102.二叉树的层次遍历(相关话题:广度优先/栈)

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

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

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

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

Java代码:

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new LinkedList<>();

        if(null != root){
            List<List<TreeNode>> tmp = new LinkedList<>();
            List<TreeNode> l0 = new LinkedList<>();
            l0.add(root);
            tmp.add(l0);
            for(int i = 0; i < tmp.size(); i++){
                List<Integer> list = new LinkedList<>();
                for(int j = 0; j < tmp.get(i).size(); j++){
                    TreeNode cur = tmp.get(i).get(j);
                    list.add(cur.val);
                    if(null != cur.left || null != cur.right){
                        if(tmp.size()-1 == i){
                            List<TreeNode> li = new LinkedList<>();
                            tmp.add(li);
                        }
                        if(null != cur.left)
                            tmp.get(i+1).add(cur.left);
                        if(null != cur.right)
                            tmp.get(i+1).add(cur.right);
                    }
                }
                res.add(list);
            }
        }

        return res;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_38823568/article/details/82982305