PAT甲级 1130 Infix Expression

题目链接：点击打开链接

思路：首先找到树根，然后根据给出的树的信息，进行中序遍历。需要注意的是括号何时输出的问题。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
string data[25];
int n,root,le[25],rig[25];
int book[25];
int findRoot(){
	for(int i = 1;i <= n;i++){
		if(!book[i]){
			return i;
		}
	}
}
void inTravel(int ro){//除了树根和叶节点，其他的子树，均要加括号 
	bool flag;
	if(ro == -1) return;
	flag = le[ro] != -1||rig[ro] != -1?true:false;
	if(ro == root) flag = false;
	if(flag) putchar('(');
	if(le[ro] != -1){
		inTravel(le[ro]);
	}
	cout<<data[ro];
	if(rig[ro] != -1){
		inTravel(rig[ro]);
	}
	if(flag) putchar(')');
	return;
}
int main(){
	scanf("%d",&n);
	memset(book,0,sizeof(book));
	for(int i = 1;i <= n;i++){
		cin>>data[i]>>le[i]>>rig[i];
		if(le[i] != -1) book[le[i]] = 1;
		if(rig[i] != -1) book[rig[i]] = 1;
	}
	root = findRoot();
	inTravel(root);
	return 0;
}