编程之美 3.9 ：重建二叉树

问题描述：给定一个前序遍历和一个中序遍历，重建一棵二叉树。如：

前序：a b d c e f

中序：d b a e c f

思路：根据前序，找出根节点a，然后根据中序，可以找出根节点a的左右子树，然后递归求a的左子树前序b d跟中序d b，a的右子树前序c e f跟中序e c f

package suda.alex.chapter3;

import java.util.Scanner;

public class RebuildTree {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        while (scanner.hasNext()) {
            int n = scanner.nextInt();
            String preStr = scanner.next();
            String inStr = scanner.next();
            char[] pre = preStr.toCharArray();
            char[] in = inStr.toCharArray();
            Node tree = reBuildTree(pre, in);
            printTree(tree);
        }
    }

    public static Node reBuildTree(char[] pre, char[] in) {
        int n = pre.length; // 二叉树的节点数
        Node root = new Node(pre[0]);
        int leftLen = 0;
        while (leftLen < n && in[leftLen] != pre[0]) { // 在中序中查找pre[0]出现的位置。注：pre[0]就是此时的根
            leftLen++;
        }
        int rightLen = n - leftLen - 1;
        char[] left1 = new char[leftLen]; // 查找到pre[0]之后，前序中的左子数序列。
        char[] right1 = new char[rightLen]; // 前序中的右子数序列
        char[] left2 = new char[leftLen]; // 中序中的左子数序列
        char[] right2 = new char[rightLen];// 中序中的右子数序列
        // 赋值
        for (int i = 1; i <= leftLen; i++) {
            left1[i - 1] = pre[i];
            left2[i - 1] = in[i - 1];
        }
        for (int i = 0; i < rightLen; i++) {
            right1[i] = pre[i + leftLen + 1];
            right2[i] = in[i + leftLen + 1];
        }
        if (leftLen > 0) {
            root.left = reBuildTree(left1, left2);
        }
        if (rightLen > 0) {
            root.right = reBuildTree(right1, right2);
        }
        return root;
    }

    // 打印出二叉树 前序 用于验证生成的二叉树是否正确
    public static void printTree(Node root) {
        if (root == null) {
            return;
        }
        System.out.print(root.val);
        printTree(root.left);
        printTree(root.right);
    }
}

class Node {
    char val;
    Node left;
    Node right;

    public Node(char val) {
        this.val = val;
    }
}