Tree，Uva 548 算法竞赛入门经典例题6-8

题目

链接

点此处跳转题目

输入

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

书上的AC代码

#include<iostream>
#include<string>
#include<sstream>//使用stringstream需要的头文件
#include<cstdio>
#include<cstring>
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);//把line给ss
	n = 0;
	int x;
	while (ss >> x) a[n++] = x;//从一行line里读入数字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];//记录根
	//cout <<"root="<< root << endl;
	int p = L1;
	//cout << "p=" << p << endl;
	while (in_order[p] != root) p++;
	int cnt = p - L1; //左子树的结点个数  
	//cout << "cnt=" << cnt << endl;
	//cout << L1 << " " << p - 1 << " " << L2 << " " << L2 + cnt - 1 << endl;
	lch[root] = build(L1, p - 1, L2, L2 + cnt - 1);//左子树建树，L1,P-1是在中序里找这段树，L2,L2+cnt-1是在后序里找
	//cout << p+1 << " " << R1 << " " << L2+cnt << " " << R2-1 << endl;
	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;
}

后记

今天还有四道例题要学，先把代码和解析丢上来，至于别的有空再说
——蒟蒻的第四篇博客
2021-1-22 16:09