目录
题目:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示
数据范围:输入二叉树的节点数 0 \le n \le 10000≤n≤1000,二叉树中每个节点的值 0\le val \le 10000≤val≤1000
要求:空间复杂度O(1)O(1)(即在原树上操作),时间复杂度 O(n)O(n)
注意:
1.要求不能创建任何新的结点,只能调整树中结点指针的指向。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode,有左右指针,其实可以看成一个双向链表的数据结构
4.你不用输出双向链表,程序会根据你的返回值自动打印输出
输入描述:
二叉树的根节点
返回值描述:
双向链表的其中一个头节点。
示例1
输入:
{10,6,14,4,8,12,16}
返回值:
From left to right are:4,6,8,10,12,14,16;From right to left are:16,14,12,10,8,6,4;
说明:
输入题面图中二叉树,输出的时候将双向链表的头节点返回即可。
解题思路:
这里我们定义两个指针,一个cur一个prev。
我们只需要保证cur指针在向后遍历的同时,能够保证两个指针互指就行了
这里先给出总代码,再解释。
总代码:
class Solution {
public:
void InOrderConvert(TreeNode* cur,TreeNode*& prev)//递归,让两个指针互指
{
if(cur==nullptr)
{
return;
}
InOrderConvert(cur->left,prev);
cur->left=prev;
if(prev)
{
prev->right=cur;
}
prev=cur;
InOrderConvert(cur->right,prev);
}
TreeNode* Convert(TreeNode* pRootOfTree) {
if(pRootOfTree==nullptr)
{
return nullptr;
}
TreeNode* prev=nullptr;
InOrderConvert(pRootOfTree,prev);
TreeNode* head=pRootOfTree;
while(head->left)
head=head->left;
return head;
}
};
图解:
(1)将根节点传入函数中,直到遇到左为空的情况
InOrderConvert(cur->left,prev);
(2)此时将prev为空,跳过if的判断句,cur直接给prev,此时prev就为4,然后4的右为空,返回6,6成为了现在的cur
(3)然后此时cur的left已经指向了prev,只需要再将prev指向cur就行了