面试题 04.06. 后继者

题目：

设计一个算法，找出二叉搜索树中指定节点的“下一个”节点（也即中序后继）。

如果指定节点没有对应的“下一个”节点，则返回null。

思路：
实际上是否为搜索树不是很要紧，对树进行中序遍历即可。利用了两个全局变量，一个用来判断是否已经找到指定节点（简单讲就是一个标志位），另一个变量来记录返回的节点指针。讲起来简单，实际还是有些坑的（先贴代码）。

代码：

int flag = 0;
struct TreeNode* ans;
void dfs (struct TreeNode* root, struct TreeNode* p)
{
    if (root)
    {
        dfs(root->left, p);
        if (flag)
        {
            ans = root;
            flag = 0;
            return;
        }
        if (root->val == p->val)    flag = 1;   
        dfs(root->right, p);
        
    }

    return ;
}


struct TreeNode* inorderSuccessor(struct TreeNode* root, struct TreeNode* p)
{
    if (!root)  return NULL;
    flag = 0;
    ans = NULL;
    dfs(root, p);
    return ans;
}

注意点：

1. 一开始写完后发现总是返回根节点，后来发现是flag的问题，找到目标节点后，每一次遍历都会执行if（flag），所以最后返回的就变成了根节点了。
2. 当然也可以利用二叉搜索树的性质：每个节点比左子树大，比右子树小。可以优化一下。