剑指offer——JZ36 二叉搜索树与双向链表 解题思路与具体代码【C++】

一、题目描述与要求

二叉搜索树与双向链表_牛客题霸_牛客网 (nowcoder.com)

题目描述

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。如下图所示

数据范围：输入二叉树的节点数 0≤n≤1000，二叉树中每个节点的值 0≤val≤1000
要求：空间复杂度O(1)（即在原树上操作），时间复杂度 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;

说明：输入题面图中二叉树，输出的时候将双向链表的头节点返回即可。

示例2：

输入：{5,4,#,3,#,2,#,1}

返回值：From left to right are:1,2,3,4,5;From right to left are:5,4,3,2,1;

说明：

                    5
                  /
                4
              /
            3
          /
        2
      /
    1
树的形状如上图  

---

二、解题思路

根据题目描述处的案例可以看出，所转换的双向链表其实就是对二叉搜索树的中序遍历的结果，只是需要我们去调整指针指向去形成相应的双向链表。

中序遍历的思路就是先将整个二叉树顺着左子树一直找到最底下，然后再回溯到上一级，然后访问右子树，然后返回上上一级，这就是递归的步骤。

而实现双向链表的连接，就需要借用一个指针pre用来指向中序遍历后当前结点的前一个节点，这样在pre还有pRootOfTree指针的配合下就能够实现双向链表的建立。

首先我们定义head指针，这是用于指向双向链表的头结点的指针（也就是顺着左子树一直到最底下的结点）；

然后定义pre指针，用于指向中序遍历序列中当前结点的前驱；

首先判断当前结点是否为空，是空的话直接返回（这也是递归结束的标志）；

然后按照中序遍历的思路，我们先递归到二叉树的最左边，然后初始化pre和head指针为最左边的结点，然后就是返回上一级，此时pre指向最左的结点，pRootOfTree指向中间节点，此时就利用这两个指针，去建立双向链表的连接【pre->left=pRootOfTree;pRootOfTree->right=pre;】，同时更新pre的指向；访问完中间节点后就是访问右子树，一样利用pre去建立连接，如此循环往复；

一直到遍历完整个二叉树后返回head即可。

---

三、具体代码

class Solution {
public:
    //返回的第一个指针，初始化为空
    TreeNode* head=nullptr;
	//中序遍历序列中当前结点的前一位，初始化为空
	TreeNode* pre=nullptr;
    TreeNode* Convert(TreeNode* pRootOfTree) {
        //空节点则返回空
		if(pRootOfTree==nullptr) return nullptr;
		//递归到最左边
		Convert(pRootOfTree->left);
		//找到第一个值，初始化head和pre
        if(pre==nullptr){
			head=pRootOfTree;
			pre=pRootOfTree;
		}
		else{
			pre->right=pRootOfTree;
			pRootOfTree->left=pre;
			pre=pRootOfTree;
		}
		Convert(pRootOfTree->right);
		return head;
    }
};