package top.actim.test09;
public class Solution {
/**
*
* @param preorderData long长整型一维数组
* @param inorderData long长整型一维数组
* @return long长整型一维数组
*/
public long[] buildPostOrder(long[] preorderData, long[] inorderData) {
// write code here
long[] ans = new long[preorderData.length];
build(ans, preorderData, inorderData, 0, 0, 0, preorderData.length);
return ans;
}
// 序列数组，后、前、中，起始指针，当前子数组长度
void build(long[] ans, long[] pre, long[] ino, int a, int p, int o, int len) {
// 前序树第一个元素为后序树最后一个元素【根左右、左右根】
long tar = pre[p];
ans[a + len - 1] = tar;
// 长度为1时都一样，基准条件
if (len == 1) {
return;
}
// 找到中序树的根【左根右】
int mid = 0;
for (int i = o; i < o + len; i++) {
if (ino[i] == tar) {
mid = i;
}
}
// 计算左右子树长度
int llen = mid - o;
int rlen = len - llen - 1;
// 递归处理左右子树
build(ans, pre, ino, a, p + 1, o, llen);
build(ans, pre, ino, a + llen, p + llen + 1, o + llen + 1, rlen);
}
// 没有全判例测试
/*
1
/ \
2 3
*/
public static void main(String[] args) {
long[] pre = {
1, 2, 3 };
long[] ino = {
2, 1, 3 };
long[] ans = new Solution().buildPostOrder(pre, ino);
for (int i = 0; i < ans.length; i++) {
System.out.print(ans[i]);
System.out.print(", ");
}
}
}
这脑子一写这种切数组的就找不好边界