-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathlevel_order_traversals.cpp
More file actions
103 lines (80 loc) · 1.38 KB
/
Copy pathlevel_order_traversals.cpp
File metadata and controls
103 lines (80 loc) · 1.38 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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#include <iostream>
#include <queue>
using namespace std;
class Node{
public:
int data;
Node *left,*right;
Node(int val){
data=val;
left=right=NULL;
}
};
void print_level_order(Node *root){
if(root==NULL)
{
return;
}
queue<Node*> q;
q.push(root);
q.push(NULL);
while(!q.empty()){
Node *node=q.front();
q.pop();
if(node!=NULL){
cout<<node->data<<" ";
if(node->left)
q.push(node->left);
if(node->right)
q.push(node->right);
}
else if(!q.empty()) {
q.push(NULL);
}
}
}
int sumNode(Node *root,int k){
if(root==NULL){
return -1;
}
queue <Node*> q;
q.push(root);
q.push(NULL);
int level=0;
int sum=0;
while(!q.empty()){
Node *node=q.front();
q.pop();
if(node!=NULL){
if(level==k){
sum+=node->data;
}
if(node->left)
q.push(node->left);
if(node->right)
q.push(node->right);
}else if(!q.empty()){
q.push(NULL);
level++;
}
}
return sum;
}
void print(Node* root){
print(root->left);
cout<<root->data<<" ";
print(root->right);
}
int main(){
Node *root=new Node(1);
root->left=new Node(2);
root->right=new Node(3);
root->left->left=new Node(4);
root->left->right=new Node(5);
root->right->left=new Node(6);
root->right->right=new Node(7);
print_level_order(root);
cout<<endl;
cout<<sumNode(root,2)<<endl;
return 0;
}