A Tour of Go - Exercise: Maps

wordcount.go

package main

import (
	"golang.org/x/tour/wc"
	"strings"
)

func WordCount(s string) map[string]int {
	words := strings.Fields(s)
	var count map[string]int
	count = make(map[string]int)
	for i := 0; i < len(words); i++ {
		v, ok := count[words[i]]
		if ok != true {
			count[words[i]] = 1
		} else {
			count[words[i]] = v + 1
		}
	}
	
	return count
}

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