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