程序员面试金典——输出单层结点

题目描述

对于一棵二叉树，请设计一个算法，创建含有某一深度上所有结点的链表。

给定二叉树的根结点指针TreeNode* root，以及链表上结点的深度，请返回一个链表ListNode，代表该深度上所有结点的值，请按树上从左往右的顺序链接，保证深度不超过树的高度，树上结点的值为非负整数且不超过100000。

思路：用层次遍历，用size控制每层的数量。每次dep--。dep从1开始的。当dep减到0的时候，构造链表 返回。

import java.util.*;

/*
public class ListNode {
    int val;
    ListNode next = null;

    ListNode(int val) {
        this.val = val;
    }
}*/
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}*/
public class TreeLevel {
    public ListNode getTreeLevel(TreeNode root, int dep) {
        // write code here
        if(root==null||dep==0)return null;
		Queue<TreeNode> queue = new LinkedList<>();
		queue.offer(root);
		while(!queue.isEmpty()){
			int size = queue.size();
			dep-=1;
			int i=1;
			if(dep==0){
					ListNode v = new ListNode(-1);
					ListNode p = v;
					while(!queue.isEmpty()){
						ListNode newNode = new ListNode(queue.poll().val);
						p.next = newNode;
						p = p.next;
					}
					return v.next;
			}else {
				while(i<=size){
					TreeNode temp=queue.poll();
					if(temp.left!=null){
						queue.offer(temp.left);
					}
					if(temp.right!=null){
						queue.offer(temp.right);
					}
                    i++;
				}
			}
			//while i
		}//queue
		return null;
    }
}