题目链接:https://ac.nowcoder.com/acm/contest/1086/A

我这种方法不是最简单的方法,不过我觉得比较好理解

题意:

给出一个01字串,每次从中间折取建成一个二叉树,若子树都是0则是B树,都是1则是I树,否则为F树,求出该树的后续遍历

思路:

首先我想到的是这个建树过程和线段树很像,每个节点存储的是以该点为根的什么样的树,0代表B树,1代表I树,2代表F树。接着我又想到了数据结构中学过的链式二叉树的后续遍历。

code:

#include <algorithm>
#include <iostream>
#include <cstring>
#include <vector>
#include <string>

#define ls node*2 + 1    // 左儿子
#define rs node*2 + 2    // 右儿子

using namespace std;
const int N = 10000;

string str;
int tree[N];
vector<int> ve;

int pow2(int n)
{
	int res = 1;
	for(int i=0; i < n; i++)
		res *= 2;
	return res;
}

void push_up(int node)
{
	// 判断子节点的类型来决定父节点的类型
	if(tree[ls] == 0 && tree[rs] == 0)
		tree[node] = 0;
	else if(tree[ls] == 1 && tree[rs] == 1)
		tree[node] = 1;
	else if(tree[ls] + tree[rs] == 1)
		tree[node] = 2;
	else
		tree[node] = 2;
}

// 建树
void build(int start, int end, int node)
{
	if(start == end)    // 不能再分了就是叶子节点
	{
		if(str[start] == '0')
			tree[node] = 0;
		else if(str[start] == '1')
			tree[node] = 1;
		return;
	}
	int mid = (start + end) / 2;    // 每次从中间折取
	build(start, mid, ls);
	build(mid + 1, end, rs);
	push_up(node);
}

// 后续遍历,左右根
void postOrder(int node)
{
	if(tree[node] == -1)    // 超出树的范围
		return;
	postOrder(ls);
	postOrder(rs);
	ve.push_back(tree[node]);	// 把遍历结果存储到vector中
	
}

int main()
{
	int n;
	cin >> n;
	cin >> str;
	ve.clear();
	memset(tree, -1, sizeof(tree));
	build(0, pow2(n)-1, 0);
	postOrder(0);
	int len = ve.size();
	for(int i=0; i < len; i++)
	{
		if(ve[i] == 0)
			cout << 'B';
		else if(ve[i] == 1)
				cout << 'I';
		else
			cout << 'F';
	}
	cout << endl;
	
	
	return 0;
}