一道比较难的题目,构建二叉树的层序遍历序列倒是不难,难在反而行之,用二叉树的层序遍历序列构建二叉树。前者不必多说。后者解题技巧为:构建一个队列,同时维护一个遍历序列指针i
。将序列第一个元素构建节点压入队列,记为head
,i = 1
。并反复进行如下操作,直到队列为空。最后返回head
即可。
- 队首出队,记作
node
- 构建
node
左子树,i++
,并且将其入队 - 构建
node
右子树,i++
,并且将其入队
算法复杂度:
- 时间复杂度: O ( n ) O(n) O(n), n n n为节点的个数,需要访问所有的节点和额外的空节点。而空节点最多有 n + 1 n+1 n+1个
- 空间复杂度: O ( n ) O(n) O(n),队列最多装 ( n + 1 ) / 2 (n + 1)/2 (n+1)/2个节点,也就是最后一层节点
public class Codec {
// Encodes a tree to a single string.
public String serialize(TreeNode root) {
Queue<TreeNode> queue = new LinkedList();
StringBuffer str = new StringBuffer();
queue.offer(root);
while(!queue.isEmpty()){
TreeNode temp = queue.poll();
if(temp != null){
str.append(temp.val);
queue.offer(temp.left);
queue.offer(temp.right);
}
else
str.append('#');
str.append(',');
}
str.deleteCharAt(str.length() - 1);
return str.toString();
}
// Decodes your encoded data to tree.
public TreeNode deserialize(String data) {
if(data.equals("#"))
return null;
// int end = data.length() - 1;
// System.out.println(data);
// while(data.charAt(end) == '#')
// end--;
String[] res = data.split(",");
// System.out.println(Arrays.toString(res));
TreeNode head = new TreeNode(Integer.valueOf(res[0]));
Queue<TreeNode> queue = new LinkedList();
queue.offer(head);
int i = 1;
while(!queue.isEmpty()){
TreeNode temp = queue.poll();
if(!res[i].equals("#")){
temp.left = new TreeNode(Integer.valueOf(res[i]));
queue.offer(temp.left);
}
i++;
if(!res[i].equals("#")){
temp.right = new TreeNode(Integer.valueOf(res[i]));
queue.offer(temp.right);
}
i++;
}
return head;
}
}
这一题也可以用先序遍历递归来解决, 但是由于最后得到的是先序遍历序列而不是层序遍历序列,而题目中的样例先序遍历序列和层序遍历序列是相同的,不过别的样例都是层序遍历序列,所以虽然能够通过测试,但是不符合题目实际的要求。
public class Codec {
// Encodes a tree to a single string.
public String serialize(TreeNode root) {
if(root == null)
return "#";
return root.val + "," + serialize(root.left) + "," + serialize(root.right);
}
// Decodes your encoded data to tree.
public TreeNode deserialize(String data) {
Queue<String> dataInQueue = new LinkedList(Arrays.asList(data.split(",")));
System.out.println(dataInQueue);
return deserialize(dataInQueue);
}
private TreeNode deserialize(Queue<String> data){
String first = data.poll();
if(first.equals("#"))
//注意这里不用用==,等于是比较hashcode,而equal是比较内容的
return null;
TreeNode node = new TreeNode(Integer.valueOf(first));
//valueOf可以把整数转化为字符串,而字符串中的该方法可以把整数转化为整数
node.left = deserialize(data);
node.right = deserialize(data);
return node;
}
}