LeetCode 99. 恢复二叉搜索树（中序遍历）

1. 题目

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

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

2. 解题

• 循环中序遍历（栈），记录不满足的节点，交换其val
• O ( n ) O(n) O(n) 空间复杂度

代码语言：javascript复制

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);
    }
};

遍历 解决方案 搜索

0 人点赞