Words count in go tour

words_count.go

package main

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

func WordCount(s string) map[string]int {
	words := strings.Fields(s)
	counts := make(map[string]int)
	for _, w := range words {
		_, ok := counts[w]
		if ok {
			counts[w] += 1
		} else {
			counts[w] = 1
		}
	}
	return counts
}