/
customs.go
70 lines (53 loc) · 1.18 KB
/
customs.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package advent2020
import (
"bufio"
"io"
)
type Group struct {
Affirmatives map[int]map[rune]struct{}
}
func ParseGroups(r io.Reader) []Group {
var groups []Group
scanner := bufio.NewScanner(r)
var groupMember int
group := Group{
Affirmatives: map[int]map[rune]struct{}{},
}
for scanner.Scan() {
affirmatives := scanner.Text()
if len(affirmatives) == 0 {
groups = append(groups, group)
group = Group{
Affirmatives: map[int]map[rune]struct{}{},
}
groupMember = 0
continue
}
if _, present := group.Affirmatives[groupMember]; !present {
group.Affirmatives[groupMember] = map[rune]struct{}{}
}
for _, question := range affirmatives {
group.Affirmatives[groupMember][rune(question)] = struct{}{}
}
groupMember++
}
groups = append(groups, group)
return groups
}
func CountAffirmatives(groups []Group) int {
var count int
for _, group := range groups {
yesesByQuestion := map[rune]int{}
for _, affirmatives := range group.Affirmatives {
for question := range affirmatives {
yesesByQuestion[question]++
}
}
for _, yeses := range yesesByQuestion {
if yeses == len(group.Affirmatives) {
count++
}
}
}
return count
}