设计一个算法,找出二叉搜索树中指定节点的“下一个”节点(也即中序后继)。
如果指定节点没有对应的“下一个”节点,则返回null。
示例 1:
输入: root = [2,1,3], p = 1
2
/
1 3
输出: 2
示例 2:
输入: root = [5,3,6,2,4,null,null,1], p = 6
5
/
3 6
/
2 4
/
1
输出: null
class Solution {
boolean find = false;
public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
if(root==null) return null;
TreeNode left = inorderSuccessor(root.left,p);
if(left!=null){ //left中找到了
return left;
}
if(find){ //标志位
find = false;
return root;
}
if(root==p){
find=true;
}
TreeNode right = inorderSuccessor(root.right,p);
return right;
}
}