【PAT】A1130 Infix Expression【树】

最新推荐文章于 2022-10-03 23:00:11 发布

昵称五个字

最新推荐文章于 2022-10-03 23:00:11 发布

阅读量631

点赞数

分类专栏： PAT

本文链接：https://blog.csdn.net/a617976080/article/details/100049208

版权

PAT 专栏收录该内容

163 篇文章 51 订阅

订阅专栏

Given a syntax tree (binary), you are supposed to output the corresponding infix expression, with parentheses reflecting the precedences of the operators.

Input Specification:

Each input file contains one test case. For each case, the first line gives a positive integer N (≤ 20) which is the total number of nodes in the syntax tree. Then N lines follow, each gives the information of a node (the i-th line corresponds to the i-th node) in the format:

data left_child right_child

where data is a string of no more than 10 characters, left_child and right_child are the indices of this node’s left and right children, respectively. The nodes are indexed from 1 to N. The NULL link is represented by −1. The figures 1 and 2 correspond to the samples 1 and 2, respectively.

在这里插入图片描述
Figure 1 Figure 2

Output Specification:

For each case, print in a line the infix expression, with parentheses reflecting the precedences of the operators. Note that there must be no extra parentheses for the final expression, as is shown by the samples. There must be no space between any symbols.

Sample Input 1:

8 7
a -1 -1
4 1

2 5
b -1 -1
d -1 -1

-1 6
c -1 -1

Sample Output 1:

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

Sample Input 2:

8
2.35 -1 -1

-1 4
% 7 8

2 3
a -1 -1
str -1 -1
871 -1 -1

Sample Output 2:

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

题意

给出一棵表达式树，自下向上合并输出表达式。

思路

先建树，然后dfs输出，每一层加上左右括号（根结点和叶子结点除外）。

代码

#include <cstdio>
#include <algorithm>
#define MAX_N 21
using namespace std;
struct Node{
    Node*left, *right;
    char data[11];
}nodes[MAX_N];
bool marked[MAX_N];
void dfs(Node* x, bool root){
    if(x == NULL){
        return;
    }
    bool notLeaf = x->left != NULL || x->right;
    if(notLeaf && !root) putchar('(');
    dfs(x->left, false);
    printf("%s", x->data);
    dfs(x->right, false);
    if(notLeaf && !root) putchar(')');
}
int main() {
    
    int N;
    scanf("%d", &N);
    fill(marked + 1, marked + N + 1, true);
    for(int i = 1, l, r; i <= N; i++){
        scanf("%s %d %d", nodes[i].data, &l, &r);
        if(l > 0){
            marked[l] = false;
            nodes[i].left = &nodes[l];
        }
        if(r > 0){
            marked[r] = false;
            nodes[i].right = &nodes[r];
        }
    }
    
    int root = 1;
    for(int i = 1; i <= N; i++){
        if(marked[i]){
            root = i;
            break;
        }
    }
    
    dfs(&nodes[root], true);
    return 0;
}