问题:
给一棵点带权(权各不相同,都是正整数)二叉树的中序和后序遍历,找一个叶子使得它到根的路径上的权和最小。如果有多解,该叶子本身的权应尽量小。
输入:
3 2 1 4 5 7 6
3 1 2 5 6 7 4
输出:
1
#include<iostream> //getline(cin, line)
#include<string>
#include<sstream> //stringstream
using namespace std;
// 因为各个结点的权值各不相同且都是正整数,直接用权值作为结点编号
const int maxv = 10000 + 10;
int in_order[maxv], post_order[maxv], lch[maxv], rch[maxv];
int n;
bool read_list(int* a) {
string line;
if (!getline(cin, line)) return false;
stringstream ss(line);
n = 0;
int x;
while (ss >> x) a[n++] = x;
return n > 0;
}
// 把in_order[L1..R1]和post_order[L2..R2]建成一棵二叉树,返回树根
int build(int L1, int R1, int L2, int R2) {
if (L1 > R1) return 0; // 空树
int root = post_order[R2];//后序遍历最后一个就是根节点的内容
int p = L1; // 中序遍历中第一个位置
while (in_order[p] != root) p++; //在中序遍历中找到根节点位置 p=3
int cnt = p - L1; // 左子树的结点个数
lch[root] = build(L1, p - 1, L2, L2 + cnt - 1);// 每一个子树左节点数组
rch[root] = build(p + 1, R1, L2 + cnt, R2 - 1);// 每一个子树右节点数组
return root;//每一个子树的根节点
}
int best, best_sum; // 目前为止的最优解和对应的权和
void dfs(int u, int sum) { //初始化最优叶子节点为根节点
sum += u;//累加得到当前权和
if (!lch[u] && !rch[u]) { // 该节点有子节点,就向下走
if (sum < best_sum || (sum == best_sum && u < best)) { //当前权和小于最优权和,或者当前权和等于最优权和且当前节点权小
best = u; //更新最优叶子
best_sum = sum; //更新最优权和
}
}
if (lch[u]) dfs(lch[u], sum); //还有左节点,递归
if (rch[u]) dfs(rch[u], sum);
}
int main() {
while (read_list(in_order)) { //中序遍历
read_list(post_order); //后序遍历
build(0, n - 1, 0, n - 1);
best_sum = 1000000000;
dfs(post_order[n - 1], 0);
cout << best << "\n";
}
return 0;
}
每一个子树左节点数组 lch
每一个子树右节点数组 rch
根节点:
\qquad
2
    
\;\;
4
\quad
7
\qquad
lch={0,0,3,0,2,0,0,5}
\qquad
rch={0,0,1,0,7,0,0,6}
根节点为2的子树,左节点为3,右节点1
根节点为7的子树,左节点为5,右节点6
根节点为4,左节点为2,右节点7