第五章 树 23 AcWing 1623. 中缀表达式

第五章 树 23 AcWing 1623. 中缀表达式

原题链接

AcWing 1623. 中缀表达式

算法标签

模拟 树的遍历

思路

1、使用数组l[i], r[i]存储第i个节点左右节点
2、由于数组l[i], r[i]存储第i个节点左右节点，因此节点l[i], r[i]非根节点， 记录l[i], r[i]非根节点， 遍历找出根节点
3、故在由根节点递归遍历树的时候，判断该节点是否有左右孩子， 返回的字符串数: 递归左子树的string值 + 该节点的string值 + 递归右子树的string值
4、如果该节点不是根节点 则在该返回值的左右两边加上括号
5、最后输出返回值

代码

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define int long long
#define x first
#define y second
#define ump unordered_map
#define pq priority_queue
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>=b;--i)
using namespace std;
typedef pair<int, int> PII;
const int N = 10005;
//int t, n, m, cnt, ans; 
int n, l[N], r[N];

bool leaf[N], f[N];
string w[N];
inline int rd(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}
string dfs(int u){
    string ls, rs;
    if(l[u]!=-1){
        ls=dfs(l[u]);
        if(!leaf[l[u]]){
            ls="("+ls+")";
        }
    }
    if(r[u]!=-1){
        rs=dfs(r[u]);
        if(!leaf[r[u]]){
            rs="("+rs+")";
        }
    }
    return ls+w[u]+rs;
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n;
	rep(i, 1, n+1){
	    cin>>w[i]>>l[i]>>r[i];
	    if(l[i]){
	        f[l[i]]=true;
	    }
	    if(r[i]){
	        f[r[i]]=true;
	    }
	    if(l[i]==-1&&r[i]==-1){
	        leaf[i]=true;
	    }
	}
	int root=1;
	while(f[root]){
	    root++;
	}
	cout<<dfs(root)<<"\n";
	return 0;
}

参考文献

AcWing 1623. 中缀表达式（PAT甲级辅导课）y总视频讲解

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈