秋招每日一题T11——笛卡尔树

题目描述

这里科普一下最小堆：从当前根节点开始，其左儿子和右儿子的值都比根节点大。
输入格式
第一行包含整数 N。

第二行包含 N 个两两不同的整数，表示原始序列。

输出格式
共一行，输出最小堆笛卡尔树的层序遍历序列。

数据范围
1≤N≤30,
原始序列中元素的取值范围 [−2147483648,2147483647]。

思路

①因此本题转化为，已知一个数字序列，能否生成它的最小堆（最小堆也是完全二叉树，因此要用树的生成方法，即dfs），并输出最小堆的层序遍历（bfs）。
②再现妙手： 参考了AcWing的题解后，发现了一个实用的工具min_element(l, r)，其中l是起始地址，r是终点地址的后一个地址，这个函数可以返回数组中的最小元素的位置（指针），因此加上*即可解除指针（即返回最小值）。
③用min_element()可以找到序列中的最小值，以最小值作为根左右开始递归构造最小堆。
④开一个队列作层序遍历即可。

代码

#include <iostream>
#include <string>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;
const int inf = 2147483647;
const int maxn = 35;
struct node {
	int l;
	int r;
	int data;
}tree[maxn];
int trlen = 0;
int n;
int s[maxn] = { 0 };
int dfs(int l, int r) {
	if (l >= r)	return -inf;	#在[l,r)建立最小堆
	int min = *min_element(s + l, s + r);
	int min_id = min_element(s + l, s + r) - s;
	int root = trlen;
	tree[trlen++] = node{
		dfs(l,min_id),dfs(min_id+1,r),min #递归建立最小堆，很好理解
	};
	return root;
}
void bfs() {
	queue<int> q;
	q.push(0);
	while (!q.empty()) {
		auto t = q.front();
		q.pop();
		cout << tree[t].data << ' ';
		int l = tree[t].l;
		int r = tree[t].r;
		if (l != -inf) {	#这里容易出错，l和r如果是合法儿子的话都是数组下标
		#因为使用数组来存的树，如果不合法的话就是-inf，即代表没有儿子。
			q.push(l);
		}
		if (r != -inf) {
			q.push(r);
		}
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> s[i];
	}
	dfs(0, n);
	bfs();
	return 0;
}