【题解】二叉搜索树与双向链表

二叉搜索树与双向链表

题目链接:二叉搜索树与双向链表

解题思路1:递归+中序遍历

首先题目最后要求的是一个的递增的双向链表,而二叉搜索树也是一类非常有特色的树,它的根节点大于所有左侧的节点,同时又小于所有右侧的节点,如果我们按照左中右去遍历这颗二叉树,恰巧得到的就是一个递增序列

题目同时要求不要创建新的节点,这样我们就需要在原有树上进行操作,树有左右节点指针,双向链表有前后两个指针,正好一一对应,我们修改指针指向,结合中序遍历,得到一颗递增的双向链表

代码如下:

cpp 复制代码
	TreeNode* head = nullptr;
	TreeNode* pre = nullptr;
    TreeNode* Convert(TreeNode* pRootOfTree) {
        if(pRootOfTree == nullptr) return nullptr;//递归的结束条件
		Convert(pRootOfTree->left);//递归到最左节点,是最小值
		if(pre == nullptr){
			//此时pRootOfTree是最左节点,是链表的head
			//初始化head和pre
			head = pRootOfTree;
			pre = pRootOfTree;
		}else{
			//pre是每一个pRootOfTree的前驱节点
			pre->right = pRootOfTree;
			pRootOfTree->left = pre;
			pre = pRootOfTree;
		}
		Convert(pRootOfTree->right);
		return head;
    }

解题思路2:非递归+栈

我们利用栈先进后出的特性,来模拟中序遍历出所有元素,先让所有左侧的元素进栈,再依次取出其父节点,再找该节点的右节点,将节点进行连接,连接方式和上一种思路一样

cpp 复制代码
    TreeNode* Convert(TreeNode* pRootOfTree) {
		if(pRootOfTree == nullptr) return nullptr;
        TreeNode* head = nullptr;
		TreeNode* pre = nullptr;
		stack<TreeNode*> s;
		while(pRootOfTree!=nullptr || !s.empty()){
			while(pRootOfTree!=nullptr){
				s.push(pRootOfTree);
				pRootOfTree = pRootOfTree->left;
			}
			pRootOfTree = s.top();
			s.pop();
			if(pre == nullptr){
				head = pRootOfTree;
				pre = pRootOfTree;
			}else{
				pre->right = pRootOfTree;
				pRootOfTree->left = pre;
				pre = pRootOfTree;
			}
			pRootOfTree = pRootOfTree->right;
		}
		return head;
    }
相关推荐
haoly198910 小时前
数据结构和算法篇-归并排序的两个视角-迭代和递归
数据结构·算法·归并排序
小梁努力敲代码10 小时前
java数据结构--List的介绍
java·开发语言·数据结构
保持低旋律节奏12 小时前
C++——list链表
c++·链表·list
Code小翊13 小时前
归并排序基础理解
数据结构·算法·排序算法
.小小陈.13 小时前
数据结构2:单链表
c语言·开发语言·数据结构·笔记·学习方法
草莓工作室13 小时前
数据结构4:线性表3-链式存储的线性表
数据结构
雾时之林13 小时前
数据结构--单链表
数据结构
Camel卡蒙13 小时前
数据结构——二叉搜索树Binary Search Tree(介绍、Java实现增删查改、中序遍历等)
java·开发语言·数据结构
2401_8414956413 小时前
【数据结构】基于Floyd算法的最短路径求解
java·数据结构·c++·python·算法··floyd
立志成为大牛的小牛13 小时前
数据结构——二十三、并查集的终极优化(王道408)
开发语言·数据结构·笔记·学习·程序人生·考研