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

题目描述

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

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

解题思路：

前序遍历二叉树，然后依次判定高度是否与这层相同（即高度是否为1），然后遍历左右子树。

注意链表用全局遍历，不然有一定错误。

而且在判定成功后，直接return，因为该层如果是，它的子树就一定不是。

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 {
     ListNode list =new ListNode(0);
    public ListNode getTreeLevel(TreeNode root, int dep) {
        // write code here
          if(root==null||dep<=0)return null;
		 ListNode head = list;
		 if(dep==1){
			 ListNode temp = new ListNode(root.val);
			 list.next = temp;
			 list = list.next;
             return head.next;
		 }
		 getLevel(root.left,dep-1);
		 getLevel(root.right, dep-1);
		 
		 return head.next;
    }
    
      public void getLevel(TreeNode root,int dep/*,ListNode list */){
		 if(root==null)return;
		 if(dep==1){
			 ListNode temp = new ListNode(root.val);
			 list.next = temp;
			 list = list.next;
             return;
		 }
		 getLevel(root.left,dep-1);
		 getLevel(root.right, dep-1);
	 }
}