Golang tour #40

go_tour_40.go

package main

import (
    "code.google.com/p/go-tour/wc"
    "strings"
)

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

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

Hi, I just came across this post, and I hope you don't mind me pointing something out. You don't need an if statement. The reason is that when a key is not present in a map, the zero value will be returned. Consequently, the if/else block can be replaced by output[b]++