152 lines
2.3 KiB
Go
152 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
_ "embed"
|
|
"fmt"
|
|
"strings"
|
|
|
|
aoc "go.sour.is/advent-of-code-2023"
|
|
)
|
|
|
|
// var log = aoc.Log
|
|
|
|
func main() { aoc.MustResult(aoc.Runner(run)) }
|
|
|
|
type result struct {
|
|
valuePT1 int
|
|
valuePT2 int
|
|
}
|
|
|
|
func (r result) String() string { return fmt.Sprintf("%#v", r) }
|
|
|
|
func run(scan *bufio.Scanner) (*result, error) {
|
|
var maps []Map
|
|
var m Map
|
|
|
|
for scan.Scan() {
|
|
text := scan.Text()
|
|
if len(text) == 0 {
|
|
maps = append(maps, m)
|
|
m = Map{}
|
|
}
|
|
|
|
m = append(m, []rune(text))
|
|
}
|
|
maps = append(maps, m)
|
|
|
|
score1 := 0
|
|
for _, m := range maps {
|
|
m = aoc.Transpose(reverse(m))
|
|
m.Sort()
|
|
score1 += m.Score()
|
|
}
|
|
|
|
score2 := 0
|
|
// type record [5]int
|
|
// var last, current record
|
|
// memo := make(map[record]int)
|
|
|
|
// for _, m := range maps {
|
|
// fmt.Println(m)
|
|
|
|
// m = aoc.Transpose(reverse(m))
|
|
|
|
// for i := 0; i < 1_000_000_000; i++ {
|
|
// m.Sort()
|
|
// current[0] = m.Score()
|
|
|
|
// m = aoc.Transpose(reverse(m))
|
|
// m.Sort()
|
|
// current[1] = m.Score()
|
|
|
|
// m = aoc.Transpose(reverse(m))
|
|
// m.Sort()
|
|
// current[2] = m.Score()
|
|
|
|
// m = aoc.Transpose(reverse(m))
|
|
// m.Sort()
|
|
// current[3] = m.Score()
|
|
|
|
// m = aoc.Transpose(reverse(m))
|
|
// current[4] = m.Score()
|
|
|
|
// if slices.Equal(last[:], current[:]) {
|
|
// break
|
|
// }
|
|
|
|
// v, ok := memo[current]
|
|
// if (i - 104) % 13 == 12 {
|
|
// fmt.Println(i, current, ok, v)
|
|
// }
|
|
// // if ok && v > 2 {
|
|
// // break
|
|
// // }
|
|
|
|
// memo[current] += 1
|
|
|
|
// last = current
|
|
// }
|
|
|
|
// score2 += m.Score()
|
|
// }
|
|
|
|
return &result{valuePT1: score1, valuePT2: score2}, nil
|
|
}
|
|
|
|
type Map [][]rune
|
|
|
|
func (m Map) String() string {
|
|
var buf strings.Builder
|
|
for _, row := range m {
|
|
buf.WriteString(string(row))
|
|
buf.WriteRune('\n')
|
|
}
|
|
buf.WriteRune('\n')
|
|
|
|
return buf.String()
|
|
}
|
|
func (m *Map) Sort() {
|
|
if m == nil {
|
|
return
|
|
}
|
|
for _, row := range *m {
|
|
base := 0
|
|
for i, r := range row {
|
|
if r == '#' {
|
|
base = i + 1
|
|
continue
|
|
}
|
|
if r == 'O' {
|
|
if base < i {
|
|
row[base], row[i] = row[i], row[base]
|
|
}
|
|
base++
|
|
}
|
|
}
|
|
}
|
|
}
|
|
func (m *Map) Score() int {
|
|
if m == nil {
|
|
return 0
|
|
}
|
|
|
|
sum := 0
|
|
max := len(*m)
|
|
for _, row := range *m {
|
|
for i, r := range row {
|
|
if r == 'O' {
|
|
sum += max - i
|
|
}
|
|
}
|
|
}
|
|
return sum
|
|
}
|
|
|
|
func reverse(m Map) Map {
|
|
for _, row := range m {
|
|
aoc.Reverse(row)
|
|
}
|
|
return m
|
|
}
|