【每日算法Day 73】学妹大半夜私聊我有空吗,然后竟然做出这种事!

简介: 二叉搜索树中的两个节点被错误地交换。请在不改变其结构的情况下,恢复这棵树。

题目链接


LeetCode 99. 恢复二叉搜索树[1]

题目描述


二叉搜索树中的两个节点被错误地交换。

请在不改变其结构的情况下,恢复这棵树。

示例1

输入
:[1,3,null,null,2]  
1  
/ 
3
\   2
输出:
[3,1,null,null,2]   3  / 1  \   2

示例2

输入:[3,1,4,null,null,2]  3 / \1   4   /  2输出:[2,1,4,null,null,3]  2 / \1   4   /  3

进阶:

image.png

题解


先不考虑空间复杂度,因为二叉搜索树的中序遍历是单调递增的,所以我们只需要求出它的中序遍历。然后两个结点被调换过位置,等价于递增序列中两个数调换了位置。那么我们只需要找出序列中第一个逆序对(前一个数)和最后一个逆序对(后一个数)就行了,然后换回它俩的位置。

但是中序遍历无论使用递归实现还是栈实现,空间复杂度都是树的高度,不是常数。

想一下为什么我们需要用递归或者栈来实现中序遍历?因为从根结点开始进入左子树之后,遍历完了左子树还需要回到根结点,然后再进入右子树继续遍历。但是如果你没有栈,你就没法从左子树回到根结点了,因为左子树中没有结点能指到根结点。

我们需要解决的就是这个问题,这里我们引入 Morris 遍历算法

下图是一个二叉搜索树例子:image.png

我们用递归来做中序遍历,访问结点的顺序是 4212346567 (这个序列就是 Morris 遍历得到的序列)。注意到非叶子结点 246 会被访问两次,第一次是作为根结点进入左子树,第二次是递归返回了,中序遍历到自己了。

那么不用递归或者栈,我们怎么在第二次成功返回根结点呢?注意到叶子结点的左右儿子都是空的,所以可以利用它们来指向根结点。

用结点 4 举个例子。首先访问 4 ,然后遍历它的左子树。左子树的最后一个遍历的结点是 3 ,那么我们就把 3 的右儿子指向根结点 4 。这样遍历完左子树之后,还能通过 3 的右儿子回到根结点 4 ,接着继续遍历右子树。

所以在第一次访问根结点时,首先要找出它左子树中最右边的那个叶子结点,把它的右儿子指向根结点。然后才能放心地递归遍历左子树,不用担心回不去啦。

具体实现的时候还有一些细节,看代码更好理解,我都写在注释里面了。

代码


c++

/** * Definition for a binary tree node. * struct TreeNode { *     int val; *     TreeNode *left; *     TreeNode *right; *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */classSolution {
public:
voidrecoverTree(TreeNode*root) {     
TreeNode*x=NULL, *y=NULL, *pre=NULL, *rightmost=NULL;  
while (root) {  
// 如果有左子树,就递归遍历左子树。 if (root->left) {    
rightmost=root->left;   
// 找出左子树的最右边一个叶子结点   while (rightmost->right&&rightmost->right!=root)
                {           
rightmost=rightmost->right;    
                }     
// 如果左子树最右边的叶子结点的右儿子是空的, // 那就说明根结点是第一次访问,那么就把它的右儿子指向根结点。              // 然后递归遍历左子树。if (rightmost->right!=root) {   
rightmost->right=root;  
root=root->left;          
// 否则的话说明根结点是第二次访问了,// 那就说明左子树已经递归完毕了,    // 那么就判断一下是否存在逆序对。  // 记得把左子树最右叶子结点的右儿子改回空指针。// 然后递归遍历右子树了。                    } else {            
if (pre&&pre->val>root->val) {    
if (x==NULL) x=pre;  
y=root;    
                    }           
rightmost->right=NULL;  
pre=root;       
root=root->right;     
                }          
// 如果没有左子树,那就直接遍历右子树,同时判断是否存在逆序对。                 } else {      
if (pre&&pre->val>root->val) {      
if (x==NULL) x=pre;      
y=root;         
                }             
pre=root;     
root=root->right; 
            }     
        }      
swap(x->val, y->val);  
    }
};

参考资料


[1]

LeetCode 99. 恢复二叉搜索树: https://leetcode-cn.com/problems/recover-binary-search-tree/

image.png

作者简介:godweiyang知乎同名华东师范大学计算机系硕士在读,方向自然语言处理与深度学习喜欢与人分享技术与知识,期待与你的进一步交流~


相关文章
|
算法 C++
【每日算法Day 73】学妹大半夜私聊我有空吗,然后竟然做出这种事!
【每日算法Day 73】学妹大半夜私聊我有空吗,然后竟然做出这种事!
132 0
|
3月前
|
机器学习/深度学习 算法 机器人
【水下图像增强融合算法】基于融合的水下图像与视频增强研究(Matlab代码实现)
【水下图像增强融合算法】基于融合的水下图像与视频增强研究(Matlab代码实现)
367 0
|
3月前
|
数据采集 分布式计算 并行计算
mRMR算法实现特征选择-MATLAB
mRMR算法实现特征选择-MATLAB
243 2
|
4月前
|
传感器 机器学习/深度学习 编解码
MATLAB|主动噪声和振动控制算法——对较大的次级路径变化具有鲁棒性
MATLAB|主动噪声和振动控制算法——对较大的次级路径变化具有鲁棒性
253 3
|
4月前
|
存储 编解码 算法
【多光谱滤波器阵列设计的最优球体填充】使用MSFA设计方法进行各种重建算法时,图像质量可以提高至多2 dB,并在光谱相似性方面实现了显著提升(Matlab代码实现)
【多光谱滤波器阵列设计的最优球体填充】使用MSFA设计方法进行各种重建算法时,图像质量可以提高至多2 dB,并在光谱相似性方面实现了显著提升(Matlab代码实现)
189 6
|
3月前
|
机器学习/深度学习 算法 机器人
使用哈里斯角Harris和SIFT算法来实现局部特征匹配(Matlab代码实现)
使用哈里斯角Harris和SIFT算法来实现局部特征匹配(Matlab代码实现)
199 8
|
3月前
|
机器学习/深度学习 算法 自动驾驶
基于导向滤波的暗通道去雾算法在灰度与彩色图像可见度复原中的研究(Matlab代码实现)
基于导向滤波的暗通道去雾算法在灰度与彩色图像可见度复原中的研究(Matlab代码实现)
214 8
|
3月前
|
机器学习/深度学习 算法 数据可视化
基于MVO多元宇宙优化的DBSCAN聚类算法matlab仿真
本程序基于MATLAB实现MVO优化的DBSCAN聚类算法,通过多元宇宙优化自动搜索最优参数Eps与MinPts,提升聚类精度。对比传统DBSCAN,MVO-DBSCAN有效克服参数依赖问题,适应复杂数据分布,增强鲁棒性,适用于非均匀密度数据集的高效聚类分析。
|
4月前
|
机器学习/深度学习 传感器 算法
【高创新】基于优化的自适应差分导纳算法的改进最大功率点跟踪研究(Matlab代码实现)
【高创新】基于优化的自适应差分导纳算法的改进最大功率点跟踪研究(Matlab代码实现)
296 14
|
3月前
|
开发框架 算法 .NET
基于ADMM无穷范数检测算法的MIMO通信系统信号检测MATLAB仿真,对比ML,MMSE,ZF以及LAMA
简介:本文介绍基于ADMM的MIMO信号检测算法,结合无穷范数优化与交替方向乘子法,降低计算复杂度并提升检测性能。涵盖MATLAB 2024b实现效果图、核心代码及详细注释,并对比ML、MMSE、ZF、OCD_MMSE与LAMA等算法。重点分析LAMA基于消息传递的低复杂度优势,适用于大规模MIMO系统,为通信系统检测提供理论支持与实践方案。(238字)

热门文章

最新文章