FBI树(C++)

题目描述

我们可以把由“0”和“1”组成的字符串分为三类：全“0”串称为 B 串，全“1”串 称为 I 串，既含“0”又含“1”的串则称为 F 串。 FBI树是一种二叉树，它的结点类型也包括F结点，B结点和I结点三种。由一个长度为 2^N 的“01”串S可以构造出一棵FBI树T，递归的构造方法如下：

T 的根结点为 R，其类型与串 S 的类型相同；

若串 S 的长度大于 1，将串 S 从中间分开，分为等长的左右子串 S1 和 S2；由左子串 S1 构造 R 的左子树 T1，由右子串 S2 构造 R 的右子树 T2。

现在给定一个长度为 2^N 的“01”串，请用上述构造方法构造出一棵FBI树，并输出它的 后序遍历2序列。

输入格式

输入第一行是一个整数N（0 <= N <= 10），第二行是一个长度为 2^N 的“01 ” 串。

输出格式

输出包括一行，这一行只包含一个字符串，即FBI 树的后序遍历序列。

样例输入

3
10001011

样例输出

IBFBBBFIBFIIIFF

提示

对于 40%的数据，N <= 2； 对于 100%的数据，N <= 10。
 

#include<bits/stdc++.h>
using namespace std;
int n;
const int maxn=600000;
char yjc[10001];
struct node {
	int id;
	char data;
	int lson, rson;
};

node tree[maxn];

void q(char x[], int y) {
	int l = strlen(x);
	if (l > 1) {
		int n = 0, m = 0;
		for (int i = 0; i < l; i++) {
			if (x[i] == '0')n++;
			if (x[i] == '1')m++;
		}
		if (n == l) {
			tree[y].data = 'B';
		} else {
			if (m == l) {
				tree[y].data = 'I';
			} else {
				if (n > 0 && m > 0) {
					tree[y].data = 'F';
				}
			}
		}
		tree[y].id = y;
		tree[y].lson = y * 2;
		tree[y].rson = y * 2 + 1;
		char a[10001], b[10001];
		strncpy(a, x, l / 2);
		strncpy(b, x + l / 2, l - l / 2);
		q(a, tree[y].lson);
		q(b, tree[y].rson);
	} else {
		if (x[0] == '0') {
			tree[y].data = 'B';
		}
		if (x[0] == '1') {
			tree[y].data = 'I';
		}
		tree[y].id = y;
		return;
	}
	if(l<=0)return;
}

void p(node x) {
	if (x.id != 0) {
		p(tree[x.lson]);
		p(tree[x.rson]);
		cout << x.data;
	}
}

int main() {
	cin >> n;
	scanf("%s", yjc);
	q(yjc, 1);
	p(tree[1]);

	return 0;
}