lavos/cap3/ex11.go

package cap3

type Candidate int // 候选人
type Vote int      // 票数

// Ex11 投票统计，有调整，这里不使用数组存储结果，使用hashmap表存结果
func Ex11(votes []Candidate, numOfCandidate int) map[Candidate]Vote {
    result := map[Candidate]Vote{}
    for _, candidate := range votes {
        // 无效的投票
        if candidate <= 0 || int(candidate) > numOfCandidate {
            continue
        }

        if _, exists := result[candidate]; !exists {
            result[candidate] = 1
        } else {
            result[candidate]++
        }
    }

    return result
}