-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
64 lines (50 loc) · 1.08 KB
/
main.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
package main
import (
"fmt"
"os"
"strconv"
"strings"
"github.com/jyggen/advent-of-go/internal/solver"
)
var eolReplacer *strings.Replacer
func init() {
eolReplacer = strings.NewReplacer("\n\n", ";", "\n", ",")
}
func main() {
p1, p2, err := solver.SolveFromFile(os.Stdin, SolvePart1, SolvePart2)
if err != nil {
panic(err)
}
fmt.Println(p1)
fmt.Println(p2)
}
func SolvePart1(input string) (string, error) {
groups := strings.Split(eolReplacer.Replace(strings.TrimSpace(input)), ";")
sum := 0
for _, g := range groups {
for i := 'a'; i <= 'z'; i++ {
if strings.ContainsRune(g, i) {
sum++
}
}
}
return strconv.Itoa(sum), nil
}
func SolvePart2(input string) (string, error) {
groups := strings.Split(eolReplacer.Replace(strings.TrimSpace(input)), ";")
sum := 0
for _, g := range groups {
size := strings.Count(g, ",")
parts := strings.SplitN(g, ",", 2)
if len(parts) == 1 {
sum += len(parts[0])
continue
}
for _, r := range parts[0] {
if c := strings.Count(parts[1], string(r)); c == size {
sum++
}
}
}
return strconv.Itoa(sum), nil
}