PTA 7-7 列出叶节点 java实现（25分）

一道简单的二叉树数据结构类型的题

思路：

首先对数据进行行处理，生成一个二叉树结点，按处理顺序存储进节点数组。

然后对节点数组进行遍历，把节点连成一个二叉树

最后层序遍历输出遇到的叶节点就可以了。

代码如下

import java.util.*;
import java.io.* ;
public class Main {
    //静态内部类--树叶结点
    static class TreeNode{
        int key ;  //当前结点的编号
        int leftName ;//当前节点左孩子的编号
        int rightName ;//当前节点右孩子的编号
        TreeNode left ;//左孩子
        TreeNode right ;//右孩子
        TreeNode father ;//父亲节点，用来判断当前结点是否是根结点

        public TreeNode(int key, int leftName, int rightName) {
            this.key = key;
            this.leftName = leftName;
            this.rightName = rightName;
        }

    }
    public static void main(String[] args) throws IOException {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(in.readLine());
        TreeNode[] arr = new TreeNode[n];
        for (int i = 0; i < n; i++) {
            String[] str = in.readLine().split(" ");
            int leftName, rightName;
            if (str[0].equals("-")) {
                leftName = i;
            } else {
                leftName = Integer.parseInt(str[0]);
            }
            if (str[1].equals("-")) {
                rightName = i;
            } else {
                rightName = Integer.parseInt(str[1]);
            }
            arr[i] = new TreeNode(i, leftName, rightName);
        }

        for (int i = 0; i < n; i++) {
            TreeNode node = arr[i];
            //对每个结点进行连树操作
            buildTree(node, arr);
        }
        TreeNode root = null;
        for (int i = 0; i < n; i++) {
            //没有父亲的结点即为根节点
            if (arr[i].father == null) {
                root = arr[i];
                break;
            }
        }
        print(root);
    }
    //层序遍历对根节点的输出是从上往下从左往右的
    public static void print(TreeNode root) {
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        boolean flag = false ; 
        while(queue.size()!=0){
            TreeNode node = queue.poll();
            if(node.right==null&&node.left==null){
                if(flag){
                    System.out.print(" ");
                }
                System.out.print(node.key);
                flag = true;
            }
            if(node.left!=null){
                queue.offer(node.left);
            }
            if(node.right!=null){
                queue.offer(node.right);
            }
        }

    }
    
    public static void buildTree(TreeNode node,TreeNode[] arr){
        if(node.leftName!=node.key){
            node.left = arr[node.leftName] ;
            arr[node.leftName].father = node ;
        }else{
            node.left = null ;
        }
        if(node.rightName!=node.key){
            node.right = arr[node.rightName] ;
            arr[node.rightName].father = node;
        }else{
            node.right = null ;
        }
    }
}

结果

测试点	提示	内存(KB)	用时(ms)	结果	得分
0		10384	60	答案正确	13 / 13
1		10280	59	答案正确	5 / 5
2		10264	61	答案正确	1 / 1
3		10252	59	答案正确	5 / 5
4		10356	59	答案正确	1 / 1