-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathRecoverBinarySearchTree.c
43 lines (33 loc) · 1.1 KB
/
RecoverBinarySearchTree.c
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
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
void transform(struct TreeNode * root, struct TreeNode ** nodoAnterior, struct TreeNode ** primeiroNodo, struct TreeNode ** segundoNodo){
if(root){
transform(root->left, nodoAnterior, primeiroNodo, segundoNodo);
if((*nodoAnterior != NULL) && (root->val <= (*nodoAnterior)->val)){
if(*primeiroNodo == NULL){
*primeiroNodo = *nodoAnterior;
*segundoNodo = root;
}else{
*segundoNodo = root;
}
}
*nodoAnterior = root;
transform(root->right, nodoAnterior, primeiroNodo, segundoNodo);
}
}
void recoverTree(struct TreeNode* root){
struct TreeNode * primeiroNodo = NULL;
struct TreeNode * segundoNodo = NULL;
struct TreeNode * nodoAnterior = NULL;
transform(root, &nodoAnterior, &primeiroNodo, &segundoNodo);
int temp = primeiroNodo->val;
primeiroNodo->val = segundoNodo->val;
segundoNodo->val = temp;
return;
}