FilesExpand file tree

 master

/

Recover.java

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

44 lines (34 loc) · 935 Bytes

 master

/

Recover.java

Top

File metadata and controls

• Code
• Blame

44 lines (34 loc) · 935 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

package Algorithms;

import Algorithms.tree.TreeNode;

public class Recover {

private TreeNode big = null;

private TreeNode small = null;

private TreeNode pre = new TreeNode(Integer.MIN_VALUE);

public void recoverTree(TreeNode root) {

if (root == null) {

return;

}

traverse(root);

int tmp = big.val;

big.val = small.val;

small.val = tmp;

}

private void traverse(TreeNode root) {

if (root == null) {

return;

}

traverse(root.left);

if (root.val < pre.val) {

if (big == null) {

big = pre;

small = root;

} else if (root.val < small.val){

small = root;

}

}

if (big == null) {

pre = root;

}

traverse(root.right);

}

}