题目
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
-
题目要求转换后的链表是排序过的,所以想到二叉搜索树的中序遍历是由小到大遍历的
-
二叉搜索树中,左子节点小于父节点,右子节点大于父节点,所以在转换成排序双向链表时,原先指向左子节点的指针对应链表中指向前一个节点的指针,原先指向右子节点的指针对应链表中指向后一个节点的指针
-
由于是循环链表, 设链表头节点 head 和尾节点 tail ,则应构建 head.left = tail 和 tail.right = head
按照中序遍历,当遍历转换到根节点时,它的左子树已经转换成一个排序的链表,并且是链表中的最后一个节点,是当前值最大的节点。
所以可以将左子树上这个最大节点和根节点进行连接,那么根节点就是当前链表的最大节点。
然后接下来遍历转换右子树时,要把根节点和右子树中最小的节点进行连接。
可以发现这就是一个递归转换的过程
class Solution {
Node head, pre;
public Node treeToDoublyList(Node root) {
if (root == null) return null;
_recursion(root);
pre.right = head;
head.left = pre; //进行头节点和尾节点的相互指向
return head;
}
public void _recursion(Node cur) {
if (cur == null) return;
// 递归左子树
_recursion(cur.left);
cur.left = pre;
if (pre == null) head = cur;
else pre.right = cur;
pre = cur; //pre指向当前的cur
_recursion(cur.right);//全部迭代完成后,pre指向双向链表中的尾节点
}
static class Node {
public int val;
public Node left;
public Node right;
public Node() {}
public Node(int _val) {
val = _val;
}
public Node(int _val,Node _left,Node _right) {
val = _val;
left = _left;
right = _right;
}
}
}
递归代码千万不要自己去一步一步地人肉傻递归,自己在脑中越遍历会感觉越混乱
public void _recursion(Node cur) {
// cur就看成二叉树的根节点
// 先遍历转换左子树
_recursion(cur.left);
// 代码走到这一步时左子树就已经全部遍历转换完毕
// 在这里需要将根节点跟左子树中最大节点进行连接
_recursion(cur.right); // 到这一步时可以看做将根节点跟右子树中最小节点连接,就把它看成右子树已经遍历转换完了,具体是怎么转换的,交给递归去处理,不要自己再去一步一步地想转换的过程
}