2022-8-16 剑指offer-二叉树

剑指 Offer II 053. 二叉搜索树中的中序后继

给定一棵二叉搜索树和其中的一个节点 p ，找到该节点在树中的中序后继。如果节点没有中序后继，请返回 null 。

节点 p 的后继是值比 p.val 大的节点中键值最小的节点，即按中序遍历的顺序节点 p 的下一个节点。

 1 /**
 2  * Definition for a binary tree node.
 3  * public class TreeNode {
 4  *     int val;
 5  *     TreeNode left;
 6  *     TreeNode right;
 7  *     TreeNode(int x) { val = x; }
 8  * }
 9  */
10 class Solution {
11     ArrayDeque<TreeNode> stack;
12     public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
13         stack=new ArrayDeque<>();
14         return dfs(root,p);
15     }
16 
17     public TreeNode dfs(TreeNode root,TreeNode p){
18         // for (TreeNode x:stack){
19         //     System.out.print(x.val+" ");
20         // }
21         if (root==null) return null;
22         if (root.val==p.val){
23             if (root.right!=null){
24                 TreeNode node=root.right;
25                 while (node.left!=null) node=node.left;
26                 return node;
27             }
28             while (!stack.isEmpty()){
29                 TreeNode t=stack.pop();
30                 if (t!=null&&t.val>p.val) return t;
31             }
32             return null;
33         }else if (root.val>p.val){
34             stack.push(root);
35             return dfs(root.left,p);
36         }else{
37             stack.push(root);
38             return dfs(root.right,p);
39         }
40     }
41 }

思路：记录当前节点和前继节点，中序遍历，或者利用搜索树的性质。