这道25分的题应该比很多30分的题还要难吧,不过不久前刷力扣,刚好做了一道中等题推理二叉树, 告诉了我们如何利用先序序列和中序序列来递归的构造或者遍历一颗树,其中这篇题解写的很棒,如何通过先序序列的根结点下标来计算对应左子树根结点和右子树根节点下标,从而实现递归,非常的巧妙。
这题完成后可以接着做1151 LCA in a Binary Tree (30),围绕中序序列“左根右”的性质来做题。
以下是本题代码:
import java.io.*;
import java.util.*;
public class Main {
static int N;
static String[] inorder;
static String[] preorder;
static List<Pair> list = new ArrayList<>();;
static Map<String, Integer> imap = new HashMap<>();
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
N = Integer.parseInt(br.readLine());
inorder = br.readLine().split("\\s+");
preorder = br.readLine().split("\\s+");
for (int i = 0; i < N; ++i) {
imap.put(inorder[i], i);
}
traverse(0, 0, N - 1, 0, 0);
System.out.print(list.get(0).data);
for (int i = 1; i < list.size(); ++i) {
System.out.print(" " + list.get(i).data);
}
}
private static class Pair {
private int pos;
private String data;
}
private static void traverse(int root, int left, int right, int h, int pos) {
if (left > right) return;
if (list.size() == h) {
Pair p = new Pair();
p.pos = pos;
p.data = preorder[root];
list.add(p);
} else {
Pair p = list.get(h);
if (pos < p.pos) {
p.pos = pos;
p.data = preorder[root];
}
}
int i = imap.get(preorder[root]);
traverse(root + 1, left, i - 1, h + 1, pos - 1);
traverse(root + i - left + 1, i + 1, right, h + 1, pos + 1);
}
}