复习系列2-DS二叉树——二叉树之父子结点

DS二叉树——二叉树之父子结点

题目描述

给定一颗二叉树的逻辑结构如下图，（先序遍历的结果，空树用字符‘0’表示，例如AB0C00D00），建立该二叉树的二叉链式存储结构。

编写程序输出该树的所有叶子结点和它们的父亲结点

 

输入

第一行输入一个整数t，表示有t个二叉树

第二行起，按照题目表示的输入方法，输入每个二叉树的先序遍历，连续输入t行

输出

第一行按先序遍历，输出第1个示例的叶子节点

第二行输出第1个示例中与叶子相对应的父亲节点

以此类推输出其它示例的结果

样例输入

3

AB0C00D00

AB00C00

ABCD0000EF000

样例输出

C D

B A

B C

A A

D F

C E

提示

 

import java.util.*;

/**
 * @author: Liu Canbin
 * @date: 2019/1/4
 */

public class Main {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        for (int i = 0; i <n ; i++) {
            String s = scanner.next();
            Tree tree = new Tree();
            tree.create(s);
            tree.levelOrderShow();
            Queue queue = tree.getQueue();
            System.out.println();
            while (!queue.isEmpty()){
                System.out.print(((Node)queue.poll()).parent.data+ " ");
            }
            System.out.println();
        }

    }
}

class Node
{
    char data;
    Node left;
    Node right;
    Node parent;
    Node(char ch){
        data =ch;
        left = null;
        right = null;
        parent = null;
    }

    public void setLeft(Node left) {
        this.left = left;
    }

    public void setRight(Node right) {
        this.right = right;
    }

    public void setParent(Node parent) {
        this.parent = parent;
    }
}

class Tree{
    Node root;
    int pos;
    String strTree;
    int sum;

    private Queue<Node> queue = new LinkedList<Node>();
    void create(String s){
        pos = 0;
        strTree = s;
        sum =0;
        root =createTree();
    }

    Node createTree(){
        Node t;
        char ch = strTree.charAt(pos++);
        if (ch=='0'){
            t = null;
        }else {
            t = new Node(ch);
            t.setLeft(createTree());
            if (t.left!=null){
                t.left.setParent(t);
            }
            t.setRight(createTree());
            if (t.right!=null){
                t.right.setParent(t);
            }
        }
        return t;
    }


    void levelOrderShow() {
        levelOrder(root);
    }

    private void levelOrder(Node t){
        if (t==null){
            return;
        }

        if (t.right==null && t.left ==null){
            queue.offer(t);
            System.out.print(t.data+" ");
        }
        levelOrder(t.left);
        levelOrder(t.right);
    }

    public Queue<Node> getQueue() {
        return queue;
    }
}