Tree

Tree

题链接：https://vjudge.net/contest/248639#problem/C

题解：给一棵点带权（权值各不相同，都是小于10000的正整数）的二叉树的中序和后序遍历，找一个叶子使得它到根的路径上的权和最小。如果有多解，该叶子本身的权应尽量小。输入中每两行表示一棵树，其中第一行为中序遍历，第二行为后序遍历。

紫书上有详细解释和代码

代码：

#include <iostream>
#include <sstream>
#include <cstring>
using namespace std;
const int maxValue = 10005;
int inOrder[maxValue], postOrder[maxValue];// 中序，后序数组
int n;// 节点个数
int lch[maxValue], rch[maxValue];// 二叉树的数组表示法
int bestSum, bestNum;// 最小权和，最小节点编号
bool readList(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;
}
// 利用inOrder[L1,...,R1],postOrder[L2,...R2]建立一颗二叉树,返回树根
// 后序的最后一个元素就是根节点，然后在中序中找到该节点
// 左边的节点都是左子树上的，右边的节点都是右子树上的
// 按照这样的形式，递归生成二叉树
int build(int L1, int R1, int L2, int R2) {
    if(L1 > R1) { // 空树
        return 0;
    }
    int root = postOrder[R2]; // 根节点
    int p = L1;// 在中序中找到根节点
    while(inOrder[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;
}
void dfs(int p, int sum) {// 深度优先遍历
    sum += p;
    if(!lch[p] && !rch[p]) { // 叶子节点
        if(sum < bestSum || (sum == bestSum && p < bestNum)) {
            bestNum = p;
            bestSum = sum;
        }
    }
    if(lch[p]) {
        dfs(lch[p], sum);
    }
    if(rch[p]) {
        dfs(rch[p], sum);
    }
}
int main() {
    while(readList(inOrder)) { // 读取中序，后序序列
        readList(postOrder);
        memset(lch, 0, sizeof(lch));  // 初始化
        memset(rch, 0, sizeof(rch));
        int root = build(0, n - 1, 0, n - 1);   // 构建二叉树
        bestSum = 100000000;
        dfs(root, 0);
        cout << bestNum << endl;
    }
    return 0;
}