淘先锋技术网

首页 1 2 3 4 5 6 7

思想:BFS遍历方法
用队列思想,存储每一层结点,第一层入队列,第二层出队列。
Queue a=new ArrayDeque<>();
判断root是否为空,不为空则入队列。
判断队列是否为空,不为空则执行循环。
定义每一层的大小n,引入一个list对象 level
出队元素存到level中。之后分别将左子树和右子树入队列。

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
     List<List<Integer>> res=new ArrayList<>();
     Queue<TreeNode> a=new ArrayDeque<>();
    if(root!=null){
        a.add(root);
    }
    while(!a.isEmpty()){
        int n=a.size();
        List<Integer> level=new ArrayList<>();
        for(int i=0;i<n;i++) {
            TreeNode node=a.poll();
            level.add(node.val);  
              if(node.left!=null)
              a.add(node.left);
               if(node.right!=null)
              a.add(node.right);
        }  
       res.add(level);
    }
    return res;
    }
}