PAT 甲级 1099 Build A Binary Search Tree (30 分) DFS BFS

原题链接： 1099 Build A Binary Search Tree (30 分)

题目描述：

题目输入包括结点个数（0是根节点），给了每个结点的左右子树的编号，-1 代表左/右子树不存在。最后一行给了一堆权值，让我们生成一颗 BST（二叉搜索树）。

分析：

根据题目除最后一行的信息，我们可以构建一颗树，根据权值我们便可生成一颗 BST。
如何生成 BST 呢，我们知道 BST 有严格的特征：

若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值
若它的右子树不空，则右子树上所有结点的值均大于或等于它的根结点的值
它的左、右子树也分别为二叉搜索树

这样的一棵树的权值的中序遍历，是非递减的。

请看，中序遍历顺序是11 25 38 42 45 58 67 73 82。
那么，我们就对这颗空树按照 L V R的顺序进行 DFS，依次填排好序的权值就好了。

DFS 建树完成之后，再对其进行 BFS 层次遍历。

满分代码：

#include <iostream>
#include <set>
#include <map>
#include <vector>
#include <cmath>
#include <unordered_map>
#include <cstring>
#include <algorithm>
#define inf 0x3f3f3f3f
typedef long long LL;
using namespace std;
const int MAXN = 1e2+10;
/*
	给定树的结构，给定一堆数值 我们按照中序遍历填入
	填入后按照层次遍历输出 
*/
int n, k;
int w[MAXN], value[MAXN];
int l[MAXN], r[MAXN];
// 中序遍历填值 
void dfs(int u) {
	if(u == -1)	return;
	dfs(l[u]);
	if(u != -1 && value[u] == -1) {
		value[u] = w[k++];
	}
	dfs(r[u]);
}

// 层序遍历输出 
/*
	我们对这棵树的结点依次BFS存入q数组中，输出的
	时候输出value[q[i]]就能输出权值了 
*/ 
void bfs(int root) {
	int st = 0, en = 0;
	int q[MAXN];
	q[0] = 0;
	while(st <= en) {
		int t = q[st++];
		if(l[t] != -1)	q[++en] = l[t];
		if(r[t] !=  -1)	q[++en] = r[t];
	}
	
	cout << value[q[0]];
	for(int i = 1; i < n; i++) {
		cout << " " << value[q[i]] ;
	}
	cout << endl;
} 

int main() {
	cin >> n;
	memset(value, -1, sizeof value);
	for(int i = 0; i < n; i++) {
		cin >> l[i] >> r[i];
	}
	for(int i = 0; i < n; i++) {
		cin >> w[i];
	}
	sort(w, w + n);
	dfs(0);
	bfs(0);
	return 0;
}