设计一个算法,找出二叉搜索树中指定节点的“下一个”节点(也即中序后继)。
如果指定节点没有对应的“下一个”节点,则返回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;
}
注意点:
- 一开始写完后发现总是返回根节点,后来发现是flag的问题,找到目标节点后,每一次遍历都会执行if(flag),所以最后返回的就变成了根节点了。
- 当然也可以利用二叉搜索树的性质:每个节点比左子树大,比右子树小。可以优化一下。