剑指Offer打卡day5——AcWing 19. 二叉树的下一个节点

【题目描述】

AcWing 19. 二叉树的下一个节点

【思路】

情况1：该节点存在右子树，那么其右孩子的最左边即是该节点的后继。如F节点的后继是H。
情况2：该节点（如B）不存在右子树，且该节点（B）是其父节点（D）的左孩子，则向上回溯其父节点，直到找到第一个其father的左孩子是它自己（C）的节点。则这个节点的父节点（F）是要找的那个节点（B）的后继。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode father;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode inorderSuccessor(TreeNode p) {
        if(p.right != null){ // 1、p有右子树，一直找到右子树的最左节点即可
            p = p.right;
            while( p.left != null) p = p.left;
            return p;
        }
        // 2、P没有右子树，回溯找father，找到第一个其左子树节点。
        while( p.father != null && p == p.father.right) p = p.father;
        return p.father;
        
        
    }
}