Cracking the coding interview--Q4.4

题目

原文：

Given a binary search tree, design an algorithm which creates a linked list of all the nodes at each depth (eg, if you have a tree with depth D, you’ll have D linked lists).

译文：

给一个二叉查找树，设计一个算法将每一层的所有结点构建为一个链表（例如：如果树深度为D，你将要构建D个链表）

解答

此题本质是BFS ,也就是说，如果已经构建了第i层结点的链表， 那么将此链表中每个结点的左右孩子结点取出，即可构建第i+1层结点的链表。代码如下：

import java.util.*;

class Q4_4{

public static LinkedList<LinkedList<TreeNode>> buildLevelList(TreeNode head){
LinkedList<LinkedList<TreeNode>> result=new LinkedList<LinkedList<TreeNode>>();
LinkedList<TreeNode> ls=new LinkedList<TreeNode>();
ls.add(head);
result.add(ls);
int level=0;
while(result.get(level).size()>0){
LinkedList<TreeNode> l=result.get(level);
Iterator<TreeNode> iterator=l.iterator();
LinkedList<TreeNode> ll=new LinkedList<TreeNode>();
while(iterator.hasNext()){
TreeNode tnode=iterator.next();
if(tnode.lchild!=null){
ll.add(tnode.lchild);
}
if(tnode.rchild!=null){
ll.add(tnode.rchild);
}
}
result.add(ll);
level++;
}
return result;
}
public static void main(String[] args){
int[] arr = new int[]{6,7,1,2,5,8,9,3,4};
TreeNode root = TreeNode.createBinaryTree(arr);
LinkedList<LinkedList<TreeNode>>  t = buildLevelList(root);
for(LinkedList<TreeNode> outEntry:t){
for(TreeNode tnode:outEntry){
System.out.print(tnode.value+" ");
}
System.out.println();
}
}
}

class TreeNode{
int value;
TreeNode lchild;
TreeNode rchild;
TreeNode parent;

public static void insert(TreeNode tnode,int x,TreeNode p){
if(tnode==null){
tnode=new TreeNode();
tnode.value=x;
tnode.lchild=null;
tnode.rchild=null;
if(p.value>x)
p.lchild=tnode;
else
p.rchild=tnode;
tnode.parent=p;
return;
}

if(x<tnode.value){
insert(tnode.lchild,x,tnode);
}else{
insert(tnode.rchild,x,tnode);
}
}

public static TreeNode createBinaryTree(int[] values){
TreeNode root=new TreeNode();
root.value=values[0];
for(int i=1;i<values.length;i++){
insert(root,values[i],root);
}
return root;
}
}

如有更好方法，还望交流！

---EOF---