题干:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
class Solution {
Node head,pre;
public Node treeToDoublyList(Node root) {
if(root==null) return root;
dfs(root);
head.left = pre;
pre.right = head;
return head;
}
public void dfs(Node cur){
if(cur==null) return;
dfs(cur.left);
if(pre==null) head = cur;
else{
pre.right = cur;
}
cur.left = pre;
pre = cur;
dfs(cur.right);
}
}
首先是采用中序遍历来实现二叉树到双向链表的转换,因为二叉树的中序遍历后是有序的。解决的方案是分两步,首先在中序遍历的递归中解决大部分二叉树节点到双向链表的转变(就是dfs(root)这一部分),第二步就是特殊化处理,因为头节点需要和尾节点相连(head.left = pre; pre.right = head)。这里需要提前说明的是,head是双向链表的头节点,也是中序遍历的时候第一个需要进行处理的节点,pre是指的前驱节点,因为我们需要实现双向,所以需要保存前驱节点。
在中序遍历的递归中:
终止条件:是否递归到叶子节点之后的节点。
本轮递归所要进行的操作:首先需要找到双向链表的头节点,具体是通过判断它是否存在一个pre节点,也就是前驱节点,来进行判断的,因为当递归到双向链表的头节点的时候,它是第一个需要处理的节点,所以此时的pre肯定是没有的,那么就可以直接把head节点设置为当前节点cur。然后需要更新pre节点。当递归到非头节点的时候,只需要设置当前节点cur的前驱节点是pre,pre的后驱节点是cur即可。
无返回值。