-
Notifications
You must be signed in to change notification settings - Fork 0
/
Gfg_easy_Insert_a_node_in_a_BST.cpp
70 lines (66 loc) · 1.79 KB
/
Gfg_easy_Insert_a_node_in_a_BST.cpp
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
class Solution
{
// private:
// void inorder(Node* node,vector<int>&vrr)
// {
// if(node ==NULL)return;
// inorder(node->left,vrr);
// vrr.push_back(node->data);
// inorder(node->right,vrr);
// }
// Node* bstUsingInorder(vector<int>&vrr,int s, int e)
// {
// if(s>e){
// return NULL;
// }
// int mid = s + (e-s)/2;
// int element = vrr[mid];
// Node* root = new Node(element);
// root->left = bstUsingInorder(vrr, s,mid-1);
// root->right = bstUsingInorder(vrr, mid+1, e);
// return root;
// }
public:
Node* insert(Node* node, int data) {
// Your code goes here
// vector<int>vrr;
// inorder(node,vrr);
// int flag=0;
// for(int i=0;i<vrr.size();i++){
// if(vrr[i]==data){
// flag =1;
// return node;
// }
// }
// if(flag==0){
// vrr.push_back(data);
// sort(vrr.begin(),vrr.end());
// }
// bstUsingInorder(vrr,0, vrr.size()-1);
// return node;
Node* ans = new Node(data);
if(node == NULL)
return ans;
Node* temp = node;
Node* prev = node;
while(temp!= NULL){
if(temp->data == data)
return node;
else if(temp->data > data){
prev = temp;
temp = temp->left;
}
else{
prev = temp;
temp = temp->right;
}
}
if(prev->data > data){
prev->left = ans;
}
else{
prev->right = ans;
}
return node;
}
};