Given a binary tree, design an algorithm which creates a linked list of all the nodes at each depth (e.g., if you have a tree with depth D, you'll have D linked lists).
Example
Example 1:
Input: {1,2,3,4}
Output: [1->null,2->3->null,4->null]
Explanation:
1
/ \
2 3
/
4
Example 2:
Input: {1,#,2,3}
Output: [1->null,2->null,3->null]
Explanation:
1
\
2
/
3
思路:层级搜索,BFS,注意用dummy node和curnode来保持住head和当前的node;
/**
* Definition of TreeNode:
* public class TreeNode {
* public int val;
* public TreeNode left, right;
* public TreeNode(int val) {
* this.val = val;
* this.left = this.right = null;
* }
* }
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
public class Solution {
/**
* @param root the root of binary tree
* @return a lists of linked list
*/
public List<ListNode> binaryTreeToLists(TreeNode root) {
List<ListNode> list = new ArrayList<ListNode>();
if(root == null) return list;
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
ListNode dumpyNode = new ListNode(0);
while(!queue.isEmpty()){
int size = queue.size();
ListNode cur = dumpyNode;
while(size > 0) {
TreeNode node = queue.poll();
size--;
ListNode listnode = new ListNode(node.val);
cur.next = listnode;
cur = listnode;
if(node.left != null) {
queue.offer(node.left);
}
if(node.right != null) {
queue.offer(node.right);
}
}
list.add(dumpyNode.next);
}
return list;
}
}