-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdiameter.cpp
More file actions
59 lines (47 loc) · 1.24 KB
/
diameter.cpp
File metadata and controls
59 lines (47 loc) · 1.24 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
#include <bits/stdc++.h>
using namespace std;
class BinaryTree
{
public:
int value;
BinaryTree*left;
BinaryTree*right;
BinaryTree(int value){
this->value = value;
left = nullptr;
right = nullptr;
}
};
int calculateHeight(BinaryTree*root)
{
if(root == NULL)
{
return 0;
}
return max(calculateHeight(root->left),calculateHeight(root->right)) + 1;
}
int calculateDiameter(BinaryTree*root)
{
if(root==NULL)
{
return 0;
}
int lheight = calculateHeight(root->left);
int rheight = calculateHeight(root->right);
int currDiameter = lheight + rheight + 1;
int ldiameter = calculateDiameter(root->left);
int rdiameter = calculateDiameter(root->right);
return max(currDiameter,max(ldiameter,rdiameter));
}
int main() {
BinaryTree*root = new BinaryTree(1);
root->left = new BinaryTree(2);
root->right = new BinaryTree(3);
root->left->left = new BinaryTree(4);
root->left->right = new BinaryTree(5);
root->right->left = new BinaryTree(6);
root->right->right = new BinaryTree(7);
root->left->left->left = new BinaryTree(8);
root->left->left->right = new BinaryTree(9);
cout<<calculateDiameter(root);
}