PAT 1174 Left-View of Binary Tree 一看就会

PAT 1174 Left-View of Binary Tree

原题链接

思路

一个简单的建树和bfs的题，题意比较模糊，其实就是从上到下找每一层最左边的结点

#include<bits/stdc++.h>
using namespace std;
const int N=30;
int n,in[N],pre[N];
vector<int> res;
unordered_map<int,int> pos,l,r;
int build(int il,int ir,int pl,int pr) {
    int root=pre[pl];
    int k=pos[root];
    if(il<k) l[root]=build(il,k-1,pl+1,pl+1+k-1-il);
    if(k<ir) r[root]=build(k+1,ir,pl+1+k-il,pr);
    return root;
}
void bfs(int u) {
    int hh=0,tt=0,q[N];
    q[0]=u;
    while(hh<=tt) {
        int head=hh,rear=tt;
        while(hh<=rear){
            int t=q[hh++];
            if(l.count(t)) q[++tt]=l[t];
            if(r.count(t)) q[++tt]=r[t];
        }
        res.push_back(q[head]);
    }
}
int main(){
    cin>>n;
    for(int i=0;i<n;i++) cin>>in[i],pos[in[i]]=i;
    for(int i=0;i<n;i++) cin>>pre[i];
    int root=build(0,n-1,0,n-1);
    bfs(root);
    cout<<res[0];
    for(int i=1;i<res.size();i++) cout<<' '<<res[i];
    return 0;
}