blob: 58842750a39569f790ed56aa2b097e915d1839e6 (
plain)
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
|
import java.lang.*;
import java.util.*;
import node.*;
class InvertTree {
public static TreeNode invertTree(TreeNode root) {
if (root.left == null || root.right == null)
return root;
TreeNode tmp = root.right;
root.right = invertTree(root.left);
root.left = invertTree(tmp);
return root;
}
public static void main(String[] args) {
TreeNode leftLeaf = new TreeNode(1);
TreeNode lMiddleLeaf = new TreeNode(3);
TreeNode rMiddleLeaf = new TreeNode(6);
TreeNode rightLeaf = new TreeNode(9);
TreeNode leftParent = new TreeNode(2, leftLeaf, lMiddleLeaf);
TreeNode rightParent = new TreeNode(7, rMiddleLeaf, rightLeaf);
TreeNode root = new TreeNode(4, leftParent, rightParent);
System.out.println(invertTree(root).val);
}
}
|