17.寻找下一个结点

题目描述

请设计一个算法，寻找二叉树中指定结点的下一个结点（即中序遍历的后继）。

给定树的根结点指针TreeNode* root和结点的值int p，请返回值为p的结点的后继结点的值。保证结点的值大于等于零小于等于100000且没有重复值，若不存在后继返回-1。

思路：中序遍历，将节点存入队列，对队列操作

代码如下：

import java.util.*;

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}*/
public class Successor {
    Queue<TreeNode> q=new LinkedList<TreeNode>();
    
    public int findSucc(TreeNode root, int p) {
        if(root==null)return -1;
       InOrder(root);
        int tmp=-1;
        while(!q.isEmpty()){
            tmp=q.poll().val;
            if(tmp==p){
                if(!q.isEmpty()){
                    return q.poll().val;
                 }else{
                    return -1;
                }
            }
        }
        return -1;
    }
    public void InOrder(TreeNode root){
        if(root.left!=null)
             InOrder(root.left);
        if(root!=null){
            q.offer(root);
        }
        if(root.right!=null)
            InOrder(root.right);
    }
}