#include <bits/stdc++.h>
using namespace std;
struct btNode {
int8_t value;
btNode *left = nullptr;
btNode *right = nullptr;
};
btNode preallocBt[40000001];
int lastNodeBt = 0;
inline btNode* get_new_btNode() {
return &preallocBt[lastNodeBt++];
}
struct linkedListNode {
int8_t val;
linkedListNode *next = nullptr;
bool inserted = false;
};
linkedListNode preallocLinked[2000001];
int lastNodeLnkd = 0;
static linkedListNode* get_new_lnkd_node() {
return &preallocLinked[lastNodeLnkd++];
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(0);
int N, K;
cin >> N >> K;
vector<int> values(N);
for (auto &i: values) cin >> i;
btNode root = btNode{30, 0, 0};
linkedListNode *head = get_new_lnkd_node();
linkedListNode *tail = head;
stack<btNode*> stck;
stck.push(&root);
for (int i = 0; i < N; i++) {
while (values[i] > stck.top()->value || (stck.top()->left != nullptr && stck.top()->right != nullptr)) {
btNode *top = stck.top();
if (top->right == nullptr && top->value != 0) {
tail->next = get_new_lnkd_node();
tail = tail->next;
tail->val = top->value - 1;
if (top->value == 1) tail->inserted = false;
else tail->inserted = true;
}
stck.pop();
}
tail->next = get_new_lnkd_node();
tail = tail->next;
tail->val = values[i];
tail->inserted = false;
while (stck.top()->value != values[i]) {
btNode *top = stck.top();
if (top->left != nullptr) {
// I have to go right (left completely explored)
top->right = get_new_btNode();
top->right->value = top->value - 1;
stck.push(top->right);
} else {
// I can go left
top->left = get_new_btNode();
top->left->value = top->value - 1;
stck.push(top->left);
}
}
stck.pop(); // node is == values[i], algorithm would never run next while loop
while (stck.size() > 1 && stck.top()->left != nullptr && stck.top()->right != nullptr) {
btNode *top = stck.top();
if (top->right == nullptr && top->value != 0) {
tail->next = get_new_lnkd_node();
tail = tail->next;
tail->val = top->value - 1;
if (top->value == 1) tail->inserted = false;
else tail->inserted = true;
}
stck.pop();
}
// maybe this works idk
}
while (!stck.empty()) {
btNode *top = stck.top();
if (top->right == nullptr && top->value != 0) {
tail->next = get_new_lnkd_node();
tail = tail->next;
tail->val = top->value - 1;
if (top->value == 1) tail->inserted = false;
else tail->inserted = true;
}
stck.pop();
}
head = head->next;
while (head != nullptr) {
cout << (int)head->val << ' ';
head = head->next;
}
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
116 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Runtime error |
107 ms |
262144 KB |
Execution killed with signal 9 |
3 |
Runtime error |
107 ms |
262144 KB |
Execution killed with signal 9 |
4 |
Runtime error |
115 ms |
262144 KB |
Execution killed with signal 9 |
5 |
Runtime error |
108 ms |
262144 KB |
Execution killed with signal 9 |
6 |
Runtime error |
103 ms |
262144 KB |
Execution killed with signal 9 |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
103 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Runtime error |
100 ms |
262144 KB |
Execution killed with signal 9 |
3 |
Runtime error |
106 ms |
262144 KB |
Execution killed with signal 9 |
4 |
Runtime error |
104 ms |
262144 KB |
Execution killed with signal 9 |
5 |
Runtime error |
103 ms |
262144 KB |
Execution killed with signal 9 |
6 |
Runtime error |
100 ms |
262144 KB |
Execution killed with signal 9 |
7 |
Runtime error |
106 ms |
262144 KB |
Execution killed with signal 9 |
8 |
Runtime error |
105 ms |
262144 KB |
Execution killed with signal 9 |
9 |
Runtime error |
109 ms |
262144 KB |
Execution killed with signal 9 |
10 |
Runtime error |
100 ms |
262144 KB |
Execution killed with signal 9 |
11 |
Runtime error |
103 ms |
262144 KB |
Execution killed with signal 9 |
12 |
Runtime error |
101 ms |
262144 KB |
Execution killed with signal 9 |
13 |
Runtime error |
103 ms |
262144 KB |
Execution killed with signal 9 |
14 |
Runtime error |
106 ms |
262144 KB |
Execution killed with signal 9 |