美文网首页程序员
二叉树的层次遍历

二叉树的层次遍历

作者: coder_flag | 来源:发表于2018-10-22 21:01 被阅读0次

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

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

       3
       / \
      9  20
        /  \
       15   7
    

    返回其层次遍历结果:

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

    解法:

    思路: 使用队列,把每一层的节点给保存起来,再逐个读出,该层节点的值保存在list中,而该层节点的左右子节点继续保存在队列中(确保左右子节点不为空再保存),直到队列为空。

    /**
     * Definition for a binary tree node.
     * public class TreeNode {
     *     int val;
     *     TreeNode left;
     *     TreeNode right;
     *     TreeNode(int x) { val = x; }
     * }
     */
    class Solution {
        public List<List<Integer>> levelOrder(TreeNode root) {
            List list = new LinkedList();
            Queue queue = new LinkedList();
            if(root!=null){
            queue.add(root);
            }else{
                return list;
            }
            while(!queue.isEmpty()){
                List list1 = new LinkedList();
                int size = queue.size();
                for(int i=0;i<size;i++){
                    TreeNode node = (TreeNode) queue.poll();
                    list1.add(node.val);
                    if(node.left!=null){
                    queue.add(node.left);
                    }
                    if(node.right!=null){
                    queue.add(node.right);
                    }
                }
                
                list.add(list1);
            }
            return list;
        }
    }
    



    解法2(递归):

    leetcode 高分解法,使用递归完成。

    class Solution {
        public List<List<Integer>> levelOrder(TreeNode root) {
            List<List<Integer>> l=new ArrayList<List<Integer>>();
            addLevel(l,0,root);
            //Collections.reverse(l);
            return l;
        }
        
        public void addLevel(List<List<Integer>> list, int level, TreeNode node){
            if(node==null)return ;
            if(list.size()-1<level)list.add(new ArrayList<Integer>());
            list.get(level).add(node.val);
            
            addLevel(list,level+1,node.left);
            addLevel(list,level+1,node.right);
        }
    }
    

    相关文章

      网友评论

        本文标题:二叉树的层次遍历

        本文链接:https://www.haomeiwen.com/subject/vednzftx.html