剑指offer 53.二叉树的中序后继
给定一棵二叉搜索树和其中的一个节点 p ,找到该节点在树中的中序后继。如果节点没有中序后继,请返回 null 。
节点 p 的后继是值比 p.val 大的节点中键值最小的节点,即按中序遍历的顺序节点 p 的下一个节点。
示例 1:
1
2
3
|
输入:root = [2,1,3], p = 1
输出:2
解释:这里 1 的中序后继是 2。请注意 p 和返回值都应是 TreeNode 类型。
|
示例 2:
1
2
3
|
输入:root = [5,3,6,2,4,null,null,1], p = 6
输出:null
解释:因为给出的节点没有中序后继,所以答案就返回 null 了。
|
提示:
1
2
3
|
树中节点的数目在范围 [1, 104] 内。
-105 <= Node.val <= 105
树中各节点的值均保证唯一。
|
注意:本题与主站 285 题相同:
https://leetcode-cn.com/problems/inorder-successor-in-bst/
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* inorderSuccessor(TreeNode* root, TreeNode* p) {
int n = 0;
dfs(root,p);
return ans;
}
TreeNode *ans = NULL;
void dfs(TreeNode *root,TreeNode *p) {
if(root==NULL) return;
dfs(root->left,p);
if(ans == NULL && root->val > p->val) {
ans = root;
return;
}
dfs(root->right,p);
}
};
|