invert binary tree

alastairruhm · Apr 10, 2018 · 53b31ae · 53b31ae
1 parent a65de9a
commit 53b31ae
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/invertBinaryTree/README.md b/invertBinaryTree/README.md
@@ -0,0 +1,3 @@
+https://leetcode.com/problems/invert-binary-tree/hints/
+
+Invert a binary tree.
diff --git a/invertBinaryTree/main.go b/invertBinaryTree/main.go
@@ -0,0 +1,28 @@
+package invertBinaryTree
+
+/**
+* Definition for a binary tree node.
+* type TreeNode struct {
+*     Val int
+*     Left *TreeNode
+*     Right *TreeNode
+* }
+ */
+
+// TreeNode definition of binary tree
+type TreeNode struct {
+	Val   int
+	Left  *TreeNode
+	Right *TreeNode
+}
+
+func invertTree(root *TreeNode) *TreeNode {
+	if root == nil {
+		return nil
+	}
+	root.Left = invertTree(root.Left)
+	root.Right = invertTree(root.Right)
+
+	root.Left, root.Right = root.Right, root.Left
+	return root
+}