C++ 二叉搜索树与双向链表_牛客题霸_牛客网

点击链接即可查看题目: 二叉搜索树与双向链表_牛客题霸_牛客网

一、题目

描述

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

数据范围：输入二叉树的节点数 0≤n≤10000≤n≤1000，二叉树中每个节点的值 0≤val≤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;

复制说明：

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

示例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
树的形状如上图  

二、解题思路以及代码

将访问的上一个结点变为当前结点的左,上一个结点的右变为当前结点 

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
	void InOrder(TreeNode* root, TreeNode*& pre)
	{
		if(nullptr == root)
			return;
		
		
		InOrder(root->left, pre);
		root->left = pre;
		if(pre)
			pre->right = root;
		pre = root;
		InOrder(root->right, pre);
		
	}
    TreeNode* Convert(TreeNode* pRootOfTree) 
	{
		TreeNode* pre = nullptr;
        InOrder(pRootOfTree, pre);
		while(pRootOfTree && pRootOfTree->left)
		{
			pRootOfTree = pRootOfTree->left;
		}
		return pRootOfTree;
    }
};