二叉树--uva548Tree 二叉树递归遍历

题目：uva548

分析：还能说什么，直接看懂了刘大爷的代码然后自己敲一遍啊。

ps:自己写的build_tree函数果然没有刘大爷的简洁，醉了。

而且啊，自己用cin.get()写的输入函数也没有刘大爷的stringstream好用啊。

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <sstream>
using namespace std;

const int maxn = 10005;
int in_order[maxn],post_order[maxn];
int l[maxn],r[maxn];
int n,ans,best_sum;

bool read_line(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;
}

int build_tree(int in_b,int in_e,int po_b,int po_e){
	int root = post_order[po_e];
	int i = 0;
	while(in_order[in_b+i] != root) i++;
	//对于下面4句，刘大爷一句 if(in_b>in_e) return 0;就解决了，跪服
	if(i>0) l[root] = build_tree(in_b,in_b+i-1,po_b,po_b+i-1);
	else l[root] = 0;    //调试的时候忘了写这一句，l数组也没有初始化，然后就。。一直崩溃
	if(in_b+i < in_e) r[root] = build_tree(in_b+i+1,in_e,po_b+i,po_e-1);
	else r[root] = 0;
	return root;
}

void dfs(int u,int sum){
	sum += u;
	if(!l[u] && !r[u]){
		if(sum < best_sum || (sum == best_sum && u < ans)){
			ans = u;
			best_sum = sum;
		}
	}
	if(l[u]) dfs(l[u],sum);
	if(r[u]) dfs(r[u],sum);
}
 
int main()
{
	while(read_line(in_order)) {
		read_line(post_order);
	    best_sum = 1e9;
	    build_tree(0,n-1,0,n-1);//建树没问题
	    dfs(post_order[n-1],0);
	    cout << ans << endl;
	}
	return 0;
}

参考：刘大爷紫书第六章第三节