【题解】【AcWing】1623. 中缀表达式

1623. 中缀表达式

原题传送：AcWing 1623. 中缀表达式

给定一个句法二叉树，请你输出相应的中缀表达式，并利用括号反映运算符的优先级。

输入格式

第一行包含整数 $N$ 表示二叉树的总结点个数。

接下来 $N$ 行，每行以下列格式给出一个结点的信息（第 $i$ 行对应于第 $i$ 个结点）：

data left_child right_child

其中data是一个长度不超过 $10$ 的字符串，left_child 和 right_child 分别是该结点的左右子结点编号。

所有结点编号从 $1$ 到 $N$ ，NULL用-1表示。

以下两个图分别对应样例 $1$ 和样例 $2$ 。

输出格式

请在一行输出中缀表达式，并利用括号反映运算符的优先级。

注意，不能有多余括号，请任何符号之间不得有空格。

数据范围

$1\le N\le 20$

输入样例1：

8
* 8 7
a -1 -1
* 4 1
+ 2 5
b -1 -1
d -1 -1
- -1 6
c -1 -1

输出样例1：

(a+b)*(c*(-d))

输入样例2：

8
2.35 -1 -1
* 6 1
- -1 4
% 7 8
+ 2 3
a -1 -1
str -1 -1
871 -1 -1

输出样例2：

(a*2.35)+(-(str%871))

思路：

dfs从根节点开始，如果不是叶结点，需要两边加上括号。

题解：

#include<bits/stdc++.h>

using namespace std;

const int N = 30;

int n;
int l[N], r[N];
string w[N];
bool has_father[N], is_leaf[N];

string dfs(int u)
{
	string left, right;
	if(l[u] != -1)
	{
		left = dfs(l[u]);
		if(!is_leaf[l[u]]) left = "(" + left + ")";
	}
	if(r[u] != -1)
	{
		right = dfs(r[u]);
		if(!is_leaf[r[u]]) right = "(" + right + ")";
	}
	return left + w[u] + right;
}

int main()
{	
	cin >> n;
	for(int i = 1; i <= n; i++)
	{
		cin >> w[i] >> l[i] >> r[i];
		if(l[i] != -1) 
			has_father[l[i]] = true;
		if(r[i] != -1) 
			has_father[r[i]] = true;
		
		if(l[i] == -1 && r[i] == -1) 
			is_leaf[i] = true;
	}	
	
	int root = 1;
	while(has_father[root]) root++;
	
	cout << dfs(root) << endl;

	return 0;
}