给你一个二叉搜索树和其中的某一个结点,请你找出该结点在树中顺序后继的节点。
结点 p 的后继是值比 p.val 大的结点中键值最小的结点。
示例 1:
2
/
1 3
输入: root = [2,1,3], p = 1
输出: 2
解析: 这里 1 的顺序后继是 2。请注意 p 和返回值都应是 TreeNode 类型。
示例 2:
5
/
3 6
/
2 4
/
1
输入: root = [5,3,6,2,4,null,null,1], p = 6
输出: null
解析: 因为给出的结点没有顺序后继,所以答案就返回 null 了。
注意:
假如给出的结点在该树中没有顺序后继的话,请返回 null
我们保证树中每个结点的值是唯一的
int get(struct TreeNode* node, int val, int *min, struct TreeNode** ret){
if(!node)
return;
if(node->val > val && node->val < *min){
*min = node->val;
*ret = node;
}
get(node->left, val, min, ret);
get(node->right, val, min, ret);
return 0;
}
struct TreeNode* inorderSuccessor(struct TreeNode* root, struct TreeNode* p) {
int min = INT_MAX;
struct TreeNode* ret = NULL;
get(root, p->val, &min, &ret);
if(min == INT_MAX)
return NULL;
return ret;
}