-
Notifications
You must be signed in to change notification settings - Fork 147
/
BST.cpp
129 lines (116 loc) Β· 2.83 KB
/
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
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
#include<iostream>
using namespace std;
class node{
public:
int data;
node* left;
node* right;
node(int data){
this->data=data;
this->left=NULL;
this->right=NULL;
}
};
node* root;
class BinarySearchTree{
public:
void insert(node* root,int* a,int i){
if(root==NULL){
return;
}
if(a[i]<root->data){
if(root->left==NULL){
node*nn=new node(a[i]);
root->left=nn;
}
else{
insert(root->left,a,i);
}
}
else{
if(root->right==NULL){
node*nn=new node(a[i]);
root->right=nn;
}
else{
insert(root->right,a,i);
}
}
}
void FindMax(node* root,node* max){
if(root!=NULL){
if(root->data > max->data){
max=root;
}
FindMax(root->right,max);
}
}
void del(node* root,int val){
if(root==NULL){
return;
}
if(root->left==NULL && root->right==NULL){
cout<<root->data<<endl;
cout<<val<<endl;
root=NULL;
return;
}
if(val==root->data){
node* max=root->left;
FindMax(root->left,max);
root->data=max->data;
if(max->left==NULL && max->right==NULL){
root->left=root->left->left;
return;
}
del(max,max->data);
return;
}
if(val<root->data){
del(root->left,val);
}
else{
del(root->right,val);
}
}
void preOrder(node* root){
if(root!=NULL){
cout<<root->data<<" ";
preOrder(root->left);
preOrder(root->right);
}
}
void postOrder(node* root){
if(root!=NULL){
postOrder(root->left);
postOrder(root->right);
cout<<root->data<<" ";
}
}
void inOrder(node*root){
if(root!=NULL){
inOrder(root->left);
cout<<root->data<<" ";
inOrder(root->right);
}
}
};
int main(){
int n;
cin>>n;
int* arr=new int[n];
for(int i=0;i<n;i++){
cin>>arr[i];
}
node* nn=new node(arr[0]);
root=nn;
BinarySearchTree bst;
for(int i=1;i<n;i++){
bst.insert(root,arr,i);
}
bst.preOrder(root);
int val;
cin>>val;
bst.del(root,val);
bst.preOrder(root);
}