UVa548 Tree(递归链表的结合)

#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;//n是用来便利数组的

bool read_list(int * a)//读取二叉树的各种便利
{
    string line;//一行行的读取
    if(!getline(cin, line)) return false;//遇到EOF 结束
    stringstream ss(line);//转化为流
    n = 0;
    int x;
    while(ss >> x) a[n++] = x;//在本函数中把每二个二叉树的便利存放到a数组内
    return n > 0;//当n小于等于0时，返回false
}

int build(int L1, int R1, int L2, int R2){//递归的方法构建二叉树
    if(L1 > R1) return 0;//当L1 大于 R1 时，构建完成
    int root = post_order[R2];//数根，等于后序便利的第一个字符
    int p = L1;//把L1给p
    while(in_order[p] != root) p++;//让L1不停的++,直到该二叉树的中序遍历等于根节点
    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，求出最优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;
}