Go Tour Map Exercised

wordcount.go

package main

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

func WordCount(s string) map[string]int {
	tokens := strings.Fields(s)
	m := make(map[string]int)
	for _, token := range tokens{
		elem, ok := m[token]
		if ok {	
			m[token] = elem + 1
		}
		if !ok {
			m[token] = 1
		}
	}
	return m
}

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