随堂习题-在二叉树中找到一个节点的后继节点

代码：

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.StreamTokenizer;
import java.util.Scanner;

public class Main{
    //????
    private static StreamTokenizer st = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));

    public static int nextInt() {
        try {
            st.nextToken();
            return (int) st.nval;
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }
    
    public static class TreeNode{
        int value;
        TreeNode left;
        TreeNode right;
        TreeNode parent;

        public TreeNode(int value) {
            this.value = value;
        }
    }
    public static void main(String[] args) {//树的建立
       // Scanner sc=new Scanner(System.in);
        int n=nextInt();
        int r=nextInt();
        TreeNode root=new TreeNode(r);
        //创建树，并返回头节点
        TreeNode head=buildTree(null);
        //获取要查找的结点的数值
        int cur=nextInt();
        TreeNode curNode=findNode(head,cur);//查找结点
        TreeNode nextNode=getNextNode(curNode);//寻找后继节点
        System.out.println(nextNode.value);

    }

    private static TreeNode getNextNode(TreeNode curNode) {
        if(curNode==null) return null;
        
        
        if(curNode.right!=null){//有右孩子，返回右子树上最左结点
            return getLeftNode(curNode.right);
        }else {//没有右孩子，向上找，某个孩子是父节点的左孩子使停，返回父节点
            TreeNode parent=curNode.parent;
            while (parent!=null && curNode!=parent.left){
                //向上找
                curNode=parent;
                parent=curNode.parent;

            }
            return parent;
        }

    }

    private static TreeNode getLeftNode(TreeNode node) {
        if(node==null) return null;
        while (node.left!=null){
            node=node.left;
        }
        return node;
    }

    private static TreeNode findNode(TreeNode head, int cur) {
        //TreeNode node=null;
        if (head==null) return null;
        //在左右子树上查找node结点
        TreeNode left = findNode(head.left, cur);
        TreeNode right = findNode(head.right, cur);
        //判断根节点是不是要查找的节点
        if(head.value==cur){
            return head;
        }else {
            return left!=null? left: right; 
        }
       
    }

    private static TreeNode buildTree(TreeNode parent) {
        int fa=nextInt();
        int left=nextInt();
        int right=nextInt();

        TreeNode node=new TreeNode(fa);
        node.parent=parent;
        if(left!=0){
            node.left=buildTree(node);
        }
        if(right!=0){
            node.right=buildTree(node);
        }

        return node;
    }
}

get到的知识点：

StreamTokenizer 默认按照空格分隔， 回车，tab是结束符。st.navl默认解析出来的格式是double

StreamTokenizer st =new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in))); 
 
st.nextToken(); // 获取下一组标记   默认是按照空格分割的   回车，tab是结束符 
int i=(int) st.nval;     //st.navl默认解析出的格式是double
 
st.nextToken();     
double j=st.nval; 
 
st.nextToken();     
String s=st.sval;

java.util.Scanner与java.io.StreamToken

Scanner通过正则表达式去解析一个一个“字节数据”，才会有他的 next()、nextInt()、nextBigDecimal()等方法，但是正则表达式的效率不高因此答题很容易Time Out。
 
StreamToken是通过分析“字符流”将字符拆分为一个个Token，StreamToken底层会去区分Token类型：word、number，与Scanner比较是不是省了很多。因此StreamToken比Scanner快很多。

StreamTokenizer与BufferedReader的使用

【Java基础】StreamTokenizer使用详解