题目来源:牛客网
题目描述:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示
注意:
1.要求不能创建任何新的结点,只能调整树中结点指针的指向。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode,有左右指针,其实可以看成一个双向链表的数据结构
4.你不用输出双向链表,程序会根据你的返回值自动打印输出
代码实现:
class Solution {
public:void InOrder(TreeNode* cur,TreeNode*& prev){if(cur==nullptr)return;InOrder(cur->left,prev);cur->left=prev;//当前节点的left指向前一个if(prev)//前一个的right指向当前节点prev->right=cur;prev=cur;InOrder(cur->right,prev);}TreeNode* Convert(TreeNode* pRootOfTree) {TreeNode* prev = nullptr;InOrder(pRootOfTree,prev);TreeNode* head = pRootOfTree;while(head && head->left){head = head->left;}return head;}
};
思路:
我们对二叉树和链表分析,我们对二叉树走中序遍历的顺序就是链表里存的顺序
因为题目要求空间复杂度的原因,我们要修改二叉树为链表
我们让二叉树的左指向前一个节点,right变为后一个节点即可
我们的prev使用了引用,原因是我们全程都只需要一个prev,否则会出现问题,prev是跟着cur变化的,大家可以画一画递归展开图进行理解
最后我们的头节点会变为链表中间的某一个节点,所以我们需要不断的找它的left,也就是前一个节点,直到找到头为止,然后返回即可