LeetCode 99. 恢复二叉搜索树(中序遍历)

1. 题目

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

请在不改变其结构的情况下,恢复这棵树。
你能想出一个只使用常数空间的解决方案吗?

2. 解题

  • 循环中序遍历(栈),记录不满足的节点,交换其val
  • O ( n ) O(n) O(n) 空间复杂度
    在这里插入图片描述
class Solution {
public:
    void recoverTree(TreeNode* root) {
    	if(root == NULL)
    		return;
        TreeNode *node1 = NULL, *node2 = NULL, *prev = NULL;
       	stack<TreeNode*> stk;
        while(root || !stk.empty())
        {
        	while(root)
        	{
        		stk.push(root);
        		root = root-> left;
        	}
        	root = stk.top();	
        	if(prev && prev->val > root->val)
        	{
        		node1 = node1 ? node1 : prev;//node1只被赋值一次
        		node2 = root;//被交换的数相邻,node2只赋值一次,否则2次
        	}
        	prev = root;
        	root = stk.top()->right;
        	stk.pop();
        }
        swap(node1->val, node2->val);
    }
};
©️2020 CSDN 皮肤主题: 代码科技 设计师:Amelia_0503 返回首页