leetcode后继者怎么实现

/** * 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       if r[i]==p{           return r[i+1]       }   }   return nil}