Invert Binary Tree in Go (GoLang).

invert-binarytree.go

package binarytree

// Tree is a binary tree.
type Tree struct {
    Root *Node
}

// Invert reverses the binary tree.
func Invert(tree *Tree) {
    _ = InvertFromNode(tree.Root)
}

// Node is a binary tree node.
type Node struct {
    Val int
    Left *Node
    Right *Node
}

// InvertFromNode reverses the children of a sub binary tree from node:
//  - children on the left become children on the right
//  - children on the right become children on the left
func InvertFromNode(node *Node) *Node {
    if node != nil {
        node.Left, node.Right = InvertFromNode(node.Right), InvertFromNode(node.Left)
    }
    return node
}