题目描述:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的循环双向链表。要求不能创建任何新的节点,只能调整树中节点指针的指向。
为了让您更好地理解问题,以下面的二叉搜索树为例:
我们希望将这个二叉搜索树转化为双向循环链表。链表中的每个节点都有一个前驱和后继指针。对于双向循环链表,第一个节点的前驱是最后一个节点,最后一个节点的后继是第一个节点。
下图展示了上面的二叉搜索树转化成的链表。“head” 表示指向链表中有最小元素的节点。
特别地,我们希望可以就地完成转换操作。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继。还需要返回链表中的第一个节点的指针。
题解:
题目描述了很多,大致意思就是:给定一颗二叉搜索树,让你根据从大到小排序,然后将排序后的结点改为双向循环链表。
题目意思了解清楚了之后,我们的思路也有了:首先,需要搞明白什么是二叉搜索树,二叉搜索树也称二叉排序树,有着左小右大的规律,所以,通过中序遍历(左中右)可以得到从小到大的有序数组,这就解决了第一个问题(从小到大排序);接下来,我们要解决的就是双向循环链表的问题,这需要定义一个前驱节点pre,将pre的右指针指向当前结点,将当前结点的左指针指向前驱结点,最后进行首尾相连
代码:
首先定义一个ArrayList数组存放中序遍历后的结点:
ArrayList<Node> nodeValues = new ArrayList<>();
然后进行中序遍历:
public static void dfs(Node root,ArrayList<Node> nodeValues){
if(root == null) return;
dfs(root.left,nodeValues);
nodeValues.add(root);
dfs(root.right,nodeValues);
}
中序遍历的代码很简单,就是左中右递归
接下来就是定义头节点head和前驱结点pre,然后不断遍历nodeValues数组,改变前驱节点的右指针指向和当前结点的左指针指向:
Node head = nodeValues.get(0);
Node pre = head;//前驱结点
for (int i = 1; i < nodeValues.size(); i++) {
pre.right = nodeValues.get(i);
nodeValues.get(i).left = pre;
pre = nodeValues.get(i);
; }
到这一步,已经将各结点链接为双向链表,但是题目要求是双向循环链表,所以还需要进一步的工作:pre跳出循环时,必然指向最后一个节点,只需要将pre的右指针指向head,head的左指针指向pre即可。
pre.right = head;
head.left = pre;
完整代码:
public static Node treeToDoublyList(Node root) {
if(root == null) return null;
ArrayList<Node> nodeValues = new ArrayList<>();
dfs(root,nodeValues);
Node head = nodeValues.get(0);
Node pre = head;//前驱结点
for (int i = 1; i < nodeValues.size(); i++) {
pre.right = nodeValues.get(i);
nodeValues.get(i).left = pre;
pre = nodeValues.get(i);
; }
pre.right = head;
head.left = pre;
return head;
}
//中序遍历
public static void dfs(Node root,ArrayList<Node> nodeValues){
if(root == null) return;
dfs(root.left,nodeValues);
nodeValues.add(root);
dfs(root.right,nodeValues);
}
总结:题目已说明二叉搜索树,所以搞明白二叉搜索树的意义就容易下手了
时间复杂度:遍历每个结点,复杂度为O(N)
空间复杂度:如果二叉搜索树变为链表形式,递归的深度为N,所以空间复杂度为O(N)