forked from Adityaranjanpatra/Btecky2
-
Notifications
You must be signed in to change notification settings - Fork 0
/
dhruv4
132 lines (90 loc) · 2.45 KB
/
dhruv4
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
#include <bits/stdc++.h>
using namespace std;
/* A binary tree node has data, pointer to left child
and a pointer to right child */
struct Node {
int data;
struct Node* left;
struct Node* right;
};
Node* newNode(int val) {
Node* temp = new Node;
temp->data = val;
temp->left = NULL;
temp->right = NULL;
return temp;
}
Node* buildTree(string str) {
if (str.length() == 0 || str[0] == 'N') return NULL;
vector<string> ip;
istringstream iss(str);
for (string str; iss >> str;) ip.push_back(str);
Node* root = newNode(stoi(ip[0]));
queue<Node*> queue;
queue.push(root);
int i = 1;
while (!queue.empty() && i < ip.size()) {
Node* currNode = queue.front();
queue.pop();
string currVal = ip[i];
// If the left child is not null
if (currVal != "N") {
currNode->left = newNode(stoi(currVal));
queue.push(currNode->left);
}
i++;
if (i >= ip.size()) break;
currVal = ip[i];
if (currVal != "N") {
currNode->right = newNode(stoi(currVal));
// Push it to the queue
queue.push(currNode->right);
}
i++;
}
return root;
}
/* Tree node structure used in the program
struct Node
{
int data;
struct Node* left;
struct Node* right;
Node(int x){
data = x;
left = right = NULL;
}
}; */
class Solution {
private:
int height(struct Node* node){
//base case
if(node == NULL) {
return 0;
}
int left = height(node ->left);
int right = height(node->right);
int ans = max(left, right) + 1;
return ans;
}
public:
pair<int,int> diameterFast(Node* root) {
//base case
if(root == NULL) {
pair<int,int> p = make_pair(0,0);
return p;
}
pair<int,int> left = diameterFast(root->left);
pair<int,int> right = diameterFast(root->right);
int op1 = left.first;
int op2 = right.first;
int op3 = left.second + right.second + 1;
pair<int,int> ans;
ans.first = max(op1, max(op2, op3));;
ans.second = max(left.second , right.second) + 1;
return ans;
}
int diameter(Node* root) {
return diameterFast(root).first;
}
};