JAVA程序设计: 二叉树的层次遍历(LeetCode:102)

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

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

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果:

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

思路:采用队列广搜一遍就行了。

class TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
	 TreeNode(int x) { val = x; }
}

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        
    	int last=0;
    	List<Integer> res=new ArrayList<Integer>();
    	List<List<Integer>> ans=new ArrayList<List<Integer>>();
    	
    	if(root==null)
    		return ans;
    	
    	Queue<Integer> q=new LinkedList<Integer>();
    	Queue<TreeNode> qe=new LinkedList<TreeNode>();
    	qe.add(root);q.add(0);
    	
    	while(!qe.isEmpty())
    	{
    		TreeNode now=qe.poll();
    		int tmp=q.poll();
    		if(tmp==last)
    			res.add(now.val);
    		else
    		{
    			last=tmp;
    			ans.add(new ArrayList<Integer>(res));
    			res=new ArrayList<Integer>();
    			res.add(now.val);
    		}
    		if(now.left!=null)
    		{
    			qe.add(now.left);
    			q.add(tmp+1);
    		}
    		if(now.right!=null)
    		{
    			qe.add(now.right);
    			q.add(tmp+1);
    		}
    	}
		ans.add(new ArrayList<Integer>(res));
		
		return ans;
    }	    	
}

 

上一篇:acwing 102 -利用二分枚举区间平均值


下一篇:LeetCode_102_二叉树的层次遍历