From 6a0b3fdc5459ad40b40209cf6798e604dd76d3e7 Mon Sep 17 00:00:00 2001 From: Michael Hunteman Date: Sun, 2 Oct 2022 20:50:02 -0500 Subject: Add Invert Tree --- InvertTree.java | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 InvertTree.java 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); + } +} -- cgit v1.2.3