题目
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表,第一个节点的前驱是最后一个节点,最后一个节点的后继是第一个节点。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/er-cha-sou-suo-shu-yu-shuang-xiang-lian-biao-lcof
解题思路
二叉搜索树的中序遍历是升序的,在遍历的过程中改变节点的左右指针,left作为pre指针,right为next指针,即可形成双向循环链表
代码实现
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
/*
// Definition for a Node.
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;
}
};
*/
class Solution {
Node head,tail;//定义头尾节点
public Node treeToDoublyList(Node root) {
if(root==null) return root;
dfs(root);
head.left=tail; //left作为pre指针
tail.right=head;//right为next指针
return head;
}
public void dfs(Node node){
//递归结束条件
if(node==null) return;
//中序遍历:左->根->右
dfs(node.left);
if(tail==null){//说明在第一个节点,保存为head
head=node;
}else{
tail.right =node; //tail保存的是上一个节点,改变右指针指向当前节点
}
node.left=tail;//当前节点左指针指向上一节点tail,形成双向链表
tail=node; // 保存当前节点,用于下层递归创建
dfs(node.right);
}
}
|