给定一个二叉查找树(什么是二叉查找树),以及一个节点,求该节点在中序遍历的后继,如果没有则返回null
样例
样例 1:
输入: {1,#,2}, node with value 1
输出: 2
解释:
1
\
2
样例 2:
输入: {2,1,3}, node with value 1
输出: 2
解释:
2
/ \
1 3
二叉树的表示
挑战
O(h),其中h是BST的高度。
注意事项
保证p是给定二叉树中的一个节点。(您可以直接通过内存地址找到p)
/**
* 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:
/*
* @param root: The root of the BST.
* @param p: You need find the successor node of p.
* @return: Successor of p.
*/
TreeNode * inorderSuccessor(TreeNode * root, TreeNode * p) {
// write your code here
vector<TreeNode*>tmp;
inorder(root,tmp);
tmp.push_back(NULL);
int l=0;
int r=tmp.size()-1;
while(l<=r)
{
int mid=l+(r-l)/2;
if(tmp[mid]==p)
{
return tmp[mid+1];
}
else if(tmp[mid]->val>p->val) r=mid-1;
else l=mid+1;
}
return NULL;
}
void inorder(TreeNode*root,vector<TreeNode*> &tmp)
{
if(root==NULL) return;
inorder(root->left,tmp);
tmp.push_back(root);
inorder(root->right,tmp);
}
};