1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
|
#include <iostream>
#include <iomanip>
#include <vector>
#include <queue>
using namespace std;
struct treeNode {
int value;
treeNode *left, *right;
treeNode(int n) : value{n}, left{}, right{} {}
};
vector<int> calc_level_order(treeNode* root) {
vector<int> level_order;
if (!root) return level_order;
queue<treeNode*> que;
que.push(root);
while (!que.empty()) {
treeNode *ptr = que.front(); que.pop();
level_order.push_back(ptr->value);
if (ptr->left) que.push(ptr->left);
if (ptr->right) que.push(ptr->right);
}
return level_order;
}
void print(treeNode* node) {
if (!node) return;
print(node->left);
cout << node->value << ' ';
print(node->right);
}
void print(treeNode* node, int indent) {
if (!node) return;
print(node->right, indent + 1);
cout << setw(indent * 4) << "" << node->value << '\n';
print(node->left, indent + 1);
}
void insert(treeNode*& root, int n) {
if (!root) root = new treeNode(n);
else {
treeNode** node = &root;
while (*node)
node = n < (*node)->value ? &(*node)->left : &(*node)->right;
*node = new treeNode(n);
}
}
int main() {
treeNode* root = nullptr;
for (int n: {6, 4, 7, 8, 5, 1, 3, 9, 0, 2}) insert(root, n);
print(root, 0); cout << '\n';
auto level_order = calc_level_order(root);
for (int n: level_order) cout << n << ' '; cout << '\n';
}
|