给出二叉树,输出中缀表达式,并用括号表示优先级
用递归来解决。根据当前遍历结点的左右子树的存在与否可以分为四种情况,其中只有三种情况有效:
- 左右子树都存在:返回"(" + 递归访问左子树 + 该结点的值 + 递归访问右子树 + “)”;
- 左空右不空:说明该结点是负号。返回"(" + 该结点的值 + 递归访问右子树 + “)”;
- 左不空右空:表达式树中不存在这种情况
- 左右全空:说明到达叶结点,值为操作数。返回该结点的值。
最后要判断输出的结果最外层的括号是否多余,多余要去掉
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 22;
bool isRoot[maxn];
struct node {
string data;
int lchild, rchild;
}Node[maxn];
string dfs(int i) {
if (Node[i].lchild == -1 && Node[i].rchild == -1) {
return Node[i].data;
}
else if (Node[i].lchild == -1 && Node[i].rchild != -1) {
return "(" + Node[i].data + dfs(Node[i].rchild) + ")";
}
else if (Node[i].lchild != -1 && Node[i].rchild != -1) {
return "(" + dfs(Node[i].lchild) + Node[i].data + dfs(Node[i].rchild) + ")";
}
}
int main() {
int n;
cin >> n;
fill(isRoot, isRoot + maxn, true);
for (int i = 1; i <= n; i++) { //输入表达式树
cin >> Node[i].data >> Node[i].lchild >> Node[i].rchild;
if(Node[i].lchild != -1)
isRoot[Node[i].lchild] = false;
if(Node[i].rchild != -1)
isRoot[Node[i].rchild] = false;
}
int root = 1; //找到根结点的下标
while (root <= n && isRoot[root] != true) {
root++;
}
string ans = dfs(root);
if (ans[0] == '(') ans = ans.substr(1, ans.size() - 2); //去掉外层的括号
cout << ans;
return 0;
}