树( Tree, UVa 548)

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

#include"pch.h"
#include<iostream>
#include<vector>
#include<queue>
#include<sstream>
#include<string>
#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;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值