刷题--程序员面试金典--面试题 04.06. 后继者（go）

面试题 04.06. 后继者

设计一个算法，找出二叉搜索树中指定节点的“下一个”节点（也即中序后继）。

如果指定节点没有对应的“下一个”节点，则返回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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/successor-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

 

func inorderSuccessor(root *TreeNode, p *TreeNode) *TreeNode {
    if root == nil {
        return nil
    }

    flag := 0
    res := &TreeNode{0,nil,nil}

    run(root,p,res,&flag)
    
    //2表示找到了后继节点
    if flag != 2 {
        return nil
    }

    return res
}

func run(root,p,res *TreeNode,flag *int) {
    if root == nil || *flag == 2  {
        return 
    }
    
    run(root.Left,p,res,flag)
    
    if *flag == 1  {
        *res = *root
        //已经找到了后继节点
        *flag = 2
        return
    }

    if root.Val == p.Val {
        //表示下一个节点为后继节点
        *flag = 1
    }

    run(root.Right,p,res,flag)
}