设计一个算法,找出二叉搜索树中指定节点的“下一个”节点(也即中序后继)。
如果指定节点没有对应的“下一个”节点,则返回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
思路:
hhhhh这题和上题一样,我们只需要把List< Integer >改成List< TreeNode >就可以完美解决(当然这种方法不是最优,但是最易理解) ,代码如下:
ps:这二叉树显示有问题…调整起来贼难受
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
int num = p.val;
List<TreeNode> list = new ArrayList<>();
f(root,list);
int i = 0;
for(i=0;i<list.size();i++)
{
if(list.get(i).val == num)
break;
}
if(i == list.size()-1)
return null;
else
return list.get(i+1);
}
public void f(TreeNode root,List<TreeNode> list)
{
if(root == null)
return;
f(root.left,list);
list.add(root);
f(root.right,list);
}
}