diff options
| -rw-r--r-- | InvertTree.java | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/InvertTree.java b/InvertTree.java new file mode 100644 index 0000000..5884275 --- /dev/null +++ b/InvertTree.java @@ -0,0 +1,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); + } +} |