输入:一颗二叉搜索树
输出:有序的循环链表
思路:采用中序遍历,对于每一个遍历的节点,将该节点的左指针指向前面一个节点,与此同时,将前一个节点的右指针指向该遍历的节点。同时注意两点:
- 没有对head节点进行赋值
- 最后一个节点的右指针没有指向第一个节点
解决方法:每次递归调用的时候,更新当前节点的右指针指向head节点,更新head节点的左指针为当前节点。如果pre节点为空,则更新head节点为当前节点。
主要代码如下:
class TreeNode{
int val;
TreeNode left;
TreeNode right;
public TreeNode(int val){
this.val=val;
left=null;
right=null;
}
}
public class TreeToDoubleList{
public TreeNode treeToDouleList(TreeNode root){
TreeNode pre=null;
TreeNode head=null;
treeToDoubleList(root,pre,head);
return head;
}
//p为当前遍历节点
public void treeToDoubleList(TreeNode p,TreeNode pre,TreeNode head){
if(p==null) return;
treeToDoubleList(p.left,pre,head);
p.left=pre;//当前节点的左指针指向前一个遍历的节点;
if(pre!=null)//如果前一个节点不为空,则前一个节点的右指针指向该遍历的节点
pre.right=p;
else//如果为空,表示pre未被初始化,则头节点指向该遍历节点。
head=p;
TreeNode right=p.right;
//将首尾相连接,建立循环
head.left=p;
p.right=head;
//进入下一个节点之前,更新pre的值
pre=p;
treeToDoubleList(right,pre,head);
}
}