题目:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-yu-shuang-xiang-lian-biao-lcof/
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
由于是二叉搜索树,我们可以直接使用 中序遍历
二叉树的中序遍历
void dfs(Node node){
if(node == null) return;
dfs(node.left); //左
System.out.print(node+","); 中
dfs(node.right); //右
}
需要在不创建新节点的情况下去调整树中的节点:
使用 pre 来引用前驱节点,head作为头节点(左子树的最左端的叶子节点,即最小值)
Node pre,head;
使用递归中序遍历树
终止条件: node 为 null,表示没有叶子节点了
在中序的过程 转换为链表
if(pre == null) head = node; //遍历到最小值,pre是null,给头节点引用
else pre.right = node; //上一个节点的right 引用现节点
node.left = pre; //现节点的left 引用 上一个节点 形成双向链表
pre = node; //将pre的引用替换为现节点,作为下一个节点的前驱节点
参考代码:
class Solution {
Node pre,head;
public Node treeToDoublyList(Node root) {
if(root == null) return null;
dfs(root);
pre.right = head;
head.left = pre;
return head;
}
void dfs(Node node){
if(node == null) return;
dfs(node.left); //左
if(pre == null) head = node;
else pre.right = node; //向右遍历 12345
node.left = pre; //向左遍历 54321
pre = node; //节点用pre保存
dfs(node.right); //右
}
}