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

如果有哪位同学幸运的看到这篇题解:恭喜你,你赚大了!你会看到一个简单理解而且实用的技巧。
乍一看,不就是前序中序建树嘛。。有什么奇怪的。如果此题常规建树,后序遍历后两个测试用例超时!。
那么怎么进行优化呢?显然 在寻找根结点时候,不用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;
}

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

发布了41 篇原创文章 · 获赞 5 · 访问量 6482
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览