给定一个二叉查找树(什么是二叉查找树),以及一个节点,求该节点在中序遍历的后继,如果没有则返回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.
*/
vector<TreeNode *> ret;
TreeNode * inorderSuccessor(TreeNode * root, TreeNode * p)
{
// write your code here
bianli(root);
int i = 0;
for(; i < ret.size(); i++)
{
if(ret[i]->val == p->val)
{
if(i == ret.size() - 1)
return NULL;
else
{
return ret[i+1];
}
}
}
return NULL;
}
void bianli(TreeNode * root)
{
if(root != NULL)
{
bianli(root->left);
ret.push_back(root);
bianli(root->right);
}
}
};