刷题记录:牛客NC16660[NOIP2004]FBI树

传送门:牛客
题目简介:

我们可以把由“0”和“1”组成的字符串分为三类：全“0”串称为B串，全“1”串称为I串，既含“0”又含“1”的串则称为F串。
FBI树是一种二叉树[1]，它的结点类型也包括F结点，B结点和I结点三种。由一个长度为2N的“01”串S可以构造出一棵FBI树T，递归的构造方法如下：
1) T的根结点为R，其类型与串S的类型相同；
2) 若串S的长度大于1，将串S从中间分开，分为等长的左右子串S1和S2；由左子串S1构造R的左子树T1，由右子串S2构造R的右子树T2。
现在给定一个长度为2N的“01”串，请用上述构造方法构造出一棵FBI树，并输出它的后序遍历[2]序列。

样例:

输入:
3
10001011
输出:
IBFBBBFIBFIIIFF

主要思路:按照规定左右建树即可每次建树时计算当前根节点的01的数量,确定当前根节点的类型;
网上大部分题解(包括洛谷和牛客都是采用边遍历边输出答案的方式进行,但我觉得既然是刷这道题,显然是刚开始学树这个数据结构,不如按部就班的建树+后序遍历的方式分开进行,为了以后学习线段树,平衡树等高级数据结构打好基础)

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
char a[1<<11+2];
//cnt主要是用来记录当前根节点的标号,l,r是当前序列的左右端点位置,用来计算01字符的数量
void build(string ss,int l,int r,int cnt) {
	//如果只剩下一个节点,说明这个节点时没有儿子的,那么就是一个根节点,直接保存即可
	if(l==r) {
		if(ss[l]=='0') a[cnt]='B';
		else a[cnt]='I';
		return ;
	}
	//用来计算根节点的FRI值
	int flag1=0,flag2=0;
	for(int i=l;i<=r;i++) {
		if(ss[i]=='0') flag1=1;
		if(ss[i]=='1') flag2=1;
	}
	if(flag1==1) {
		if(flag2==1) a[cnt]='F';
		else a[cnt]='B';
	}else {
		a[cnt]='I';
	}
	//建立左右子树
	build(ss,l,(l+r)/2,2*cnt+1);
	build(ss,(l+r)/2+1,r,2*cnt+2);
}
void houxubianli(char tree[],int l,int r,int cnt) {
//因为是后序遍历,此时我们也需要保留l,r的值用来判断是不是只剩单个节点
//如果是先序遍历,因为输出在遍历之前,可以不需要l,r,直接输出即可
	if(l==r) cout<<tree[cnt];
	if(r>l) {
		houxubianli(tree,l,(l+r)/2,2*cnt+1);
		houxubianli(tree,(l+r)/2+1,r,2*cnt+2);
		cout<<tree[cnt];
	}
}
int main() {
	int n;string s;
	cin>>n>>s;
	int cnt=0;
	build(s,0,pow(2,n)-1,cnt);
	houxubianli(a,0,pow(2,n)-1,cnt);
	return 0;
}