浙江大学PAT甲级A1138(C++)题解

本文介绍了一种使用map优化前序中序建树的方法,避免了寻找根节点时的for循环,显著提高了构建二叉树的效率,特别是在大规模数据集上。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

如果有哪位同学幸运的看到这篇题解:恭喜你,你赚大了!你会看到一个简单理解而且实用的技巧。
乍一看,不就是前序中序建树嘛。。有什么奇怪的。如果此题常规建树,后序遍历后两个测试用例超时!。
那么怎么进行优化呢?显然 在寻找根结点时候,不用for循环(题中注释),而改用map!map在输入的时候已经记录了根节点的位置!

#include <iostream>
#include <vector>
#include <map>
using namespace std;
struct Node{
    int data;
    Node* lchild;
    Node* rchild;
};
int pre[100100];
int in[100100];
vector<int> v;
map<int,int> m;
Node* create(int preL,int preR,int inL,int inR){
    if(preL>preR){
        return NULL;
    }
    Node* root = new Node;
    root->data = pre[preL];
//    int k ;
//    for(k =inL;k<inR;k++){
//        if(in[k]==pre[preL])
//        break;
//    }
     int k =m[root->data];
    int numLeft = k -inL;
//    int numRight = inR-k;
 //   if(numLeft==0&&numRight==0){
  //      v.push_back(root->data);
 //       return 0;
 //   }
    root->lchild= create(preL+1,preL+numLeft,inL,k-1);
    root->rchild = create(preL+numLeft+1,preR,k+1,inR);
    return root;
}

void postOrder(Node* root){
    if(root==NULL){
        return;
    }
    postOrder(root->lchild);
    postOrder(root->rchild);

    int k=root->data;
////    cout<<root->data;
    v.push_back(k);
    return;
}
int N;
int main() {
    cin>>N;
    for(int i =0;i<N;i++){
        cin>>pre[i];

    }
    for(int i =0;i<N;i++){
        cin>>in[i];
        m[in[i]] =i;

    }

    Node* root = create(0,N-1,0,N-1);

    postOrder(root);
    cout<<v[0];

    return 0;
}

如果有哪位同学幸运的看到这篇题解:恭喜你,你赚大了!你会看到一个简单理解而且实用的技巧。
乍一看,不就是前序中序建树嘛。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值