题目:
给一棵点带权(权值各不相同,都是小于10000的正整数)的二叉树的中序和后序遍
历,找一个叶子使得它到根的路径上的权和最小。如果有多解,该叶子本身的权应尽量小。
输入中每两行表示一棵树,其中第一行为中序遍历,第二行为后序遍历。
样例输入:
3 2 1 4 5 7 6
3 1 2 5 6 7 4
7 8 11 3 5 16 12 18
8 3 11 7 16 18 12 5
255
255
样例输出:
1
3
255
分析与解答
1.先序遍历:根左右。中序遍历:左根右。后序遍历:左右根
2.通过递归把每个根连的子树分别输入到lch[],rch[]
中序:3,2,1,4,5,7,6
后序:3,1,2,5,6,7,4
根节点4
左子树a元素:3,2,1
同时左子树a的后序是:3,1,2
因此左子树a根节点是:2
左子树b元素:3
左子树b根节点:3
左子树c的最右边元素下标为b根节点下标减一,但此时b下标为零,所以最右边元素下标为-1.我们认为最左子树c最左边元素下标为0,显然此时l1>r1不成立,递归结束,我们返回0
lch[3]=0 ——> lch[2]=3——>lch[4]=2
lch[root]=leftleaf
表示一个根root所连的左子叶
rch[root]=rightleaf
表示一个根root所连的右子叶
3.叶子到根上路径权和最小,如果有多解,叶子本身权应尽量小
lch[root]=leftleaf,而每一个子叶都相当于一个子树的根,因此我们可以从根到子叶进行遍历,每次sum加上子叶的值,然后进行限制条件判断
4.输入存到两个数组里的方法
由于输入是不定长的,所以利用string,然后stringstream重定向输入到数组里
5.代码参照刘汝佳,以后复习再重新写一遍
// 题意:给一棵点带权(权各不相同,都是正整数)二叉树的中序和后序遍历,找一个叶子使得它到根的路径上的权和最小。如果有多解,该叶子本身的权应尽量小
// 算法:递归建树,然后DFS。注意,直接递归求结果也可以,但是先建树的方法不仅直观,而且更好调试
#include<iostream>
#include<string>
#include<sstream>
#include<algorithm>
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++;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;
}