30分钟,过样例,最近都在整理过去的错题,来有意识的去记忆一些常考的模板,等考好我会把最后整理好的笔记发出来的
#include <bits/stdc++.h>
using namespace std;
struct node
{
int val;
node* left;
node* right;
int level;
node() {
left = right = NULL;
}
};
vector<node> ans;
vector<int> in;
vector<int> pre;
int n;
node* build(int root, int start, int end) {
if (start > end || root > n) {
return NULL;
}
int i;
for (i = 0; i < n; i++) {
if (in[i] == pre[root]) {
break;
}
}
node* temp = new node();
temp->val = in[i];
temp->left = build(root + 1, start, i - 1);
temp->right = build(root + 1 + i - start, i + 1, end);
return temp;
}
void level_order(node* root) {
queue<node*> q;
root->level = 1;
q.push(root);
while (!q.empty()) {
node* first = q.front();
ans.push_back(*first);
q.pop();
if (first->left != NULL) {
first->left->level = first->level+1;
q.push(first->left);
}
if (first->right != NULL) {
first->right->level = first->level+1;
q.push(first->right);
}
}
}
int main()
{
#ifndef ONLINE_JUDGE
FILE* s;
freopen_s(&s, "in.txt", "r", stdin);
#endif // !ONLINE_JUDGE
cin >> n;
in.resize(n);
pre.resize(n);
int t;
for (int i = 0; i < n; i++) {
cin >> in[i];
}
for (int i = 0; i < n; i++) {
cin >> pre[i];
}
node* root = build(0, 0, n - 1);
level_order(root);
bool chance = true;
int match = 0;
for (int i = 0; i < ans.size(); i++) {
if (ans[i].level != match && chance == true) {
cout << ans[i].val;
chance = false;
match = ans[i].level;
}
else if (ans[i].level != match && chance == false) {
cout <<" "<< ans[i].val;
match = ans[i].level;
}
}
cout << endl;
return 0;
}