golang刷leetcode 技巧（24）后继者

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func inorderSuccessor(root *TreeNode, p *TreeNode) *TreeNode {
   if root==nil || p==nil{
       return nil
   }

   if root.Val<=p.Val{
       return inorderSuccessor(root.Right,p)
   }
    
    p0:=inorderSuccessor(root.Left,p)
     if p0!=nil{
       return p0
     }
     return root
}

func inorderSuccessor(root *TreeNode, p *TreeNode) *TreeNode {
   if root==nil || p==nil{
       return nil
   }
   var q []*TreeNode
   var r []*TreeNode
   for len(q)>0 || root!=nil{
       for root!=nil && root.Left!=nil{
           q=append(q,root)
           root=root.Left
       }
       fmt.Println(len(q),root,root==nil)
       if root==nil{
            l:=len(q)
            if l>0{
                root=q[l-1]
                q=q[:l-1:l-1]
                 //fmt.Println(len(q),root,root==nil)
                   r=append(r,root)
                    root=root.Right

            }
       }else{
       r=append(r,root)
        //fmt.Println(len(q),root,root.Right)
        root=root.Right
      }
   }
   for i:=0;i<len(r)-1;i++{
       if r[i]==p{
           return r[i+1]
       }
   }
   return nil
}