二叉树的递归遍历 ——树（UVa548）

//#include <bits/stdc++.h>
#include <iostream>
#include <stdio.h>
#include <string>
#include <map>
#include <set>
#include <sstream>
#include <vector>
#include <algorithm>
#include <cstring>
#include <queue>
#include <iomanip>
#include <cmath>
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;
}