程序员代码面试指南刷题--第三章.二叉树的序列化

题目描述
二叉树被记录为文件的过程叫作二叉树的序列化,通过文件内容重建原来二叉树的过程叫作二叉树的反序列化,给定一颗二叉树,请将该二叉树先序序列化和层序序列化。
假设序列化的结果字符串为 str,初始时 str = “”,遍历二叉树时,遇到 null 节点,则在 str 的末尾加上 “#!”,否则加上"当前的节点值!"。
输入描述:

第一行输入两个整数 n 和 root,n 表示二叉树的总节点个数,root 表示二叉树的根节点。
以下 n 行每行三个整数 fa,lch,rch,表示 fa 的左儿子为 lch,右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子,rch同理)

输出描述:

输出两行分别表示该二叉树的先序序列化和层序序列化

示例1

输入

2 1
1 2 0
2 0 0

输出

1!2!#!#!#!
1!2!#!#!#!

解法一:就前序和层序遍历即可

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        br.readLine();
        TreeNode root = createTree(br);
        StringBuilder sb1 = new StringBuilder();
        pre(root,sb1);
        StringBuilder sb2 = new StringBuilder();
        layer(root,sb2);
        System.out.println(sb1.toString().trim());
        System.out.println(sb2.toString().trim());
    }
    public static void layer(TreeNode root,StringBuilder sb){
        if(root==null){
            sb.append("#!");
            return;
        }
        Queue<TreeNode> q = new LinkedList<>();
        q.add(root);
        int presize = 1;
        int size = 0;
        while(!q.isEmpty()){
            TreeNode node = q.poll();
            if(node==null){
                sb.append("#!");    
            }else{
                sb.append(node.val+"!");
                q.add(node.left);
                q.add(node.right);
            }
        }
    }
    public static void pre(TreeNode root,StringBuilder sb){
        if(root==null) {
            sb.append("#!");
            return;
        }
        sb.append(root.val+"!");
        pre(root.left,sb);
        pre(root.right,sb);
    }
    public static TreeNode createTree(BufferedReader br){
        try{
            String[] arr = br.readLine().trim().split(" ");
            int data = Integer.parseInt(arr[0]);
            int left = Integer.parseInt(arr[1]);
            int right = Integer.parseInt(arr[2]);
            TreeNode root = new TreeNode(data);
            if(left!=0){
                root.left=createTree(br);
            }
            if(right!=0){
                root.right=createTree(br);
            }
            return root;
        }catch(Exception e){
            return null;
        }
    }
}
class TreeNode{
    int val;
    TreeNode left;
    TreeNode right;
    public TreeNode(int val){
        this.val = val;
    }
}
发布了189 篇原创文章 · 获赞 8 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_44406146/article/details/105449161