团体程序设计天梯赛(L2-011 玩转二叉树 (25 分))

题目：

思路分析：

数据结构的题目

只要有二种思路吧！

1.用列表存储整个树 然后递归建树 bfs输出就行 和之前的一道题目一起 刷过的应该知道！

2.用数组来存整个树 然后跟进节点的特性2*i 2*i+1的关系递归建树

代码实现：

方法1:

const int MAX=1010;
struct node{
    int lson,rson;
}e[MAX];
int mid[MAX];
int pre[MAX];
int built(int prel,int prer,int midl,int midr){
    if(midl>midr) return 0;
    int root=pre[prel];
    int i=midl;
    while (root!=mid[i]) {
        i++;
    }
    int len=i-midl;
    e[root].rson=built(prel+1,prel+len,midl,i-1);
    e[root].lson=built(prel+1+len,prer,i+1,midr);
    return root;
}
int n;
void bfs(int x){
    queue<int>q;
    q.push(x);
    vector<int>v;
    while (!q.empty()) {
        int u=q.front();
        q.pop();
        if(!u) break;
        v.push_back(u);
        if(e[u].lson) q.push(e[u].lson);
        if(e[u].rson) q.push(e[u].rson);
    }
    int len=v.size();
        for(int i=0;i<len;i++)
            printf("%d%c",v[i],i==len-1?'\n':' ');
        return;
    
}
int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>mid[i];
    }
    for(int i=0;i<n;i++){
        cin>>pre[i];
    }
    int root=pre[0];
    built(0, n-1, 0, n-1);
    bfs(root);
    return 0;
    
    
}

方法二：

const int MAX=1000010;
int n;
int mid[MAX];
int pre[MAX];
int tree[MAX];
void dfs(int pos,int lp,int rp,int lm,int rm){
    if(lm>rm) return;
    int root=pre[lp];
    int j=lm;
    tree[pos]=root;
    while (root!=mid[j]) {
        j++;
    }
    int len=j-lm;
    dfs(pos*2+1,lp+1,lp+len,lm,j-1);
    dfs(pos*2,lp+1+len,rp,j+1,rm);
}
int main(){
    int n; scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&mid[i]);
    for(int i=1;i<=n;i++) scanf("%d",&pre[i]);
    dfs(1,1,n,1,n);
    for(int i=1,j=1;i<=n;i++){
        while(tree[j]==0) j++;
        printf(i==n?"%d":"%d ",tree[j++]);
    }
    return 0;

}