A Tour of Go - Exercise: Maps

exercise-maps.go

// Implement WordCount. It should return a map of the counts of each “word” in the string s. 
// The wc.Test function runs a test suite against the provided function and prints success or failure.
// 
// You might find strings.Fields helpful.
package main

import (
	"strings"

	"golang.org/x/tour/wc"
)

func WordCount(s string) map[string]int {
	word_count_map := make(map[string]int)

	words := strings.Fields(s)
	for _, word := range words {
		element, _ := word_count_map[word]

		word_count := element + 1
		word_count_map[word] = word_count
	}

	return word_count_map
}

func main() {
	wc.Test(WordCount)
}