-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathbst_one_q.cpp
More file actions
80 lines (70 loc) · 1.63 KB
/
bst_one_q.cpp
File metadata and controls
80 lines (70 loc) · 1.63 KB
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#include<iostream>
#include <queue>
typedef struct Node {
int val;
struct Node *left;
struct Node *right;
}Node;
void levelOrder_two_q(Node *root)
{
printf("\nlevel order : ");
if(root)
{
std::queue<Node*> aQueue, bQueue;
std::queue<Node*> ¤tQueue = aQueue;
std::queue<Node*> &childQueue = bQueue;
currentQueue.push(root);
while(!currentQueue.empty())
{
root = currentQueue.front();
printf("%d ", root->val);
currentQueue.pop();
if(root->left)
childQueue.push(root->left);
if(root->right)
childQueue.push(root->right);
if(currentQueue.empty())
{
printf("\n");
swap(currentQueue, childQueue);
}
}
}
}
/**
* with single queue
*/
void levelOrder(Node *root)
{
printf("\nlevel order : ");
if(root)
{
std::queue<Node*> currentQueue;
int currentLevelNodes = 0, nextLevelNodes = 0;
currentQueue.push(root);
currentLevelNodes++;
while(!currentQueue.empty())
{
root = currentQueue.front();
printf("%d ", root->val);
currentQueue.pop();
currentLevelNodes--;
if(root->left)
{
currentQueue.push(root->left);
nextLevelNodes++;
}
if(root->right)
{
currentQueue.push(root->right);
nextLevelNodes++;
}
if(!currentLevelNodes)
{
printf("\n");
currentLevelNodes = nextLevelNodes;
nextLevelNodes = 0;
}
}
}
}