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