LeetCode 99. Recover Binary Search Tree

中序遍历查找二叉树，应获得一个单调递增的数列。

依题意，树中有两个节点的val被相互调换了。

那么在中序遍历或的数列中，应该会出现两次相邻元素，前一个节点值(pre->val)比后一个(cur->val)大的情况。

有一个特殊情况是，被题目相互交换的两个val本身就是相邻元素，这样的话只会出现一次pre->val > cur->val.

在这里只使用了常数空间的变量，但是由于递归的产生，空间复杂度的最坏情况还是O(n). 有更好的方法可以留言告诉我，谢谢。

代码：

class Solution 
{
public:
    void recoverTree(TreeNode *root) 
    {
        s1 = s2 = pre = NULL;
        gao(root);
        swap(s1->val, s2->val);
    }

private:
	void gao(TreeNode *node)
	{	
		if (node == NULL)
		{
			return ;
		}
		gao(node->left);
		if (pre!=NULL && pre->val>node->val)
		{
			if (s1 == NULL)
			{
				s1 = pre;
				s2 = node;
			} else
			{
				s2 = node;
			}
		}
		pre = node;
		gao(node->right);
	}
	TreeNode *s1, *s2, *pre;
};