[Lintcode]Binary Tree Level Order Traversal

最新推荐文章于 2024-08-23 15:23:16 发布

青铁

最新推荐文章于 2024-08-23 15:23:16 发布

阅读量183

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： lintcode

本文链接：https://blog.youkuaiyun.com/jc69186918/article/details/52205360

算法专栏收录该内容

191 篇文章

订阅专栏

本文介绍了一种解决二叉树层次遍历问题的方法。利用队列实现广度优先搜索(BFS)，通过加入虚拟节点来区分不同的层级，并返回从左到右、按层收集的节点值列表。

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

Example

Given binary tree {3,9,20,#,#,15,7},

return its level order traversal as:

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

使用队列，BFS，加入dummy节点，每遇到一次相当于换一层。使用DFS的话，可以创建辅助函数，辅助函数接受一个level参数记录层数。另外用hashmap储存level和对应节点的list

public class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: Level order a list of lists of integer
     */
    public ArrayList<ArrayList<Integer>> levelOrder(TreeNode root) {
        LinkedList<TreeNode> list = new LinkedList<TreeNode>();
        TreeNode dummy = new TreeNode(0);
        list.add(root);
        list.add(dummy);
        
        ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
        if(root == null) return res;
        
        ArrayList<Integer> line = new ArrayList<Integer>();
        while(list.size() != 1) {
            TreeNode tmp = list.poll();
            if(tmp.val == 0) {
                res.add(line);
                line = new ArrayList<Integer>();
                list.add(dummy);
            } else {
                line.add(tmp.val);
                if(tmp.left != null) list.add(tmp.left);
                if(tmp.right != null) list.add(tmp.right);
            }
        }
        res.add(line);//最后一层在这里加
        return res;
    }