add huffman encoding in go

contents/huffman_encoding/code/golang/huffman.go

@@ -0,0 +1,143 @@
+package main
+
+import (
+	"container/heap"
+	"fmt"
+)
+
+type node struct {
+	freq  int
+	char  rune
+	left  *node
+	right *node
+}
+
+type codebook map[rune]string
+type nodeHeap []*node
+
+func (n nodeHeap) Len() int           { return len(n) }
+func (n nodeHeap) Less(i, j int) bool { return n[i].freq > n[j].freq }
+func (n nodeHeap) Swap(i, j int)      { n[i], n[j] = n[j], n[i] }
+
+func (n *nodeHeap) Push(x interface{}) {
+	if node, ok := x.(*node); ok {
+		*n = append(*n, node)
+	} else {
+		fmt.Printf("I got a node of Type %T\n", x)
+	}
+}
+
+func (n *nodeHeap) Pop() interface{} {
+	old := *n
+	l := len(old)
+	x := old[l-1]
+	*n = old[0 : l-1]
+	return x
+}
+
+func buildTree(message string) *node {
+	freqMap := make(map[rune]*node)
+	h := new(nodeHeap)
+	heap.Init(h) // really needed?
+
+	for _, char := range message {
+		if _, ok := freqMap[char]; ok {
+			freqMap[char].freq++
+		} else {
+			newNode := new(node)
+			newNode.freq = 1
+			newNode.char = char
+			freqMap[char] = newNode
+			heap.Push(h, newNode)
+		}
+	}
+
+	for h.Len() > 1 {
+		left, right := h.Pop().(*node), h.Pop().(*node)
+		branch := new(node)
+		branch.freq = right.freq + left.freq
+		branch.left = left
+		branch.right = right
+		heap.Push(h, branch)
+	}
+
+	root := heap.Pop(h).(*node)
+	return root
+}
+
+func codebookRecurse(node *node, cb *codebook, code []rune) {
+	if node == nil {
+		return
+	}
+
+	if node.left == nil && node.right == nil {
+		(*cb)[node.char] = string(code)
+	}
+
+	code = append(code, '0')
+	codebookRecurse(node.left, cb, code)
+	code = append(code[:len(code)-1], '1')
+	codebookRecurse(node.right, cb, code)
+}
+
+func encode(message string) (string, *node, codebook) {
+	ret := ""
+	root := buildTree(message)
+	cb := generateCodebook(root)
+	for _, char := range message {
+		ret += cb[char]
+	}
+
+	return ret, root, cb
+}
+
+func decode(message string, root *node) string {
+	cur := root
+	ret := ""
+
+	for _, char := range message {
+		if cur == nil {
+			return message
+		}
+
+		switch string(char) {
+		case "0":
+			if cur.left == nil {
+				ret += string(cur.char)
+				cur = root.left
+			} else {
+				cur = cur.left
+			}
+		case "1":
+			if cur.right == nil {
+				ret += string(cur.char)
+				cur = root.right
+			} else {
+				cur = cur.right
+			}
+		}
+	}
+
+	if cur.char != 0 {
+		ret += string(cur.char)
+	}
+
+	return ret
+}
+
+func generateCodebook(root *node) codebook {
+	cb := make(codebook)
+	codeArr := make([]rune, 0)
+	codebookRecurse(root, &cb, codeArr)
+	return cb
+}
+
+func main() {
+	enc, root, cb := encode("bibbity_bobbity")
+	fmt.Println("Codebook:")
+	for r, c := range cb {
+		fmt.Println(string(r), "->", c)
+	}
+	fmt.Println("\nEncoded:", enc)
+	fmt.Println("Decoded:", decode(enc, root))
+}

contents/huffman_encoding/huffman_encoding.md

@@ -77,6 +77,8 @@ Whether you use a stack or straight-up recursion also depends on the language, b
 [import, lang:"javascript"](code/javascript/huffman.js)
 {% sample lang="java" %}
 [import, lang:"java"](code/java/huffman.java)
+{% sample lang="go" %}
+[import, lang:"go"](code/golang/huffman.go)
 {% sample lang="asm-x64" %}
 [import, lang:"asm-x64"](code/asm-x64/huffman.s)
 {% endmethod %}