xuu/advent-of-code
1
0
Fork 0
Code Pull Requests 1 Actions Activity

chore: add day 3 part 2

Browse Source
This commit is contained in:
xuu 2023-12-03 10:34:13 -07:00
parent 5bfb45e08f
commit c894853cbd
Signed by: xuu
GPG Key ID: 8B3B0604F164E04F
1 changed files with 28 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
day03/main.go
View File

@ -20,14 +20,20 @@ type partNumber struct {
} }
type symbol struct { type symbol struct {
symbol rune
row int row int
col int col int
adjacentParts []*partNumber
} }
type symbolTab map[int]map[int]symbol type symbolTab map[int]map[int]*symbol
func hasSymbol(tab symbolTab, row, col int) bool { func hasSymbol(tab symbolTab, row, col int, p partNumber) bool {
if cols, ok := tab[row]; ok { if cols, ok := tab[row]; ok {
_, ok = cols[col] s, ok := cols[col]
if ok {
s.adjacentParts = append(s.adjacentParts, &p)
cols[col] = s
}
return ok return ok
} }
return false return false
@ -38,7 +44,7 @@ func scanSymbol(tab symbolTab, p partNumber) bool {
for i := rowStart; i <= rowEnd; i++ { for i := rowStart; i <= rowEnd; i++ {
for j := colStart; j <= colEnd; j++ { for j := colStart; j <= colEnd; j++ {
ok := hasSymbol(tab, i, j) ok := hasSymbol(tab, i, j, p)
// fmt.Println(p.number, i, j, ok) // fmt.Println(p.number, i, j, ok)
if ok { if ok {
return true return true
@ -54,7 +60,8 @@ func main() {
m := [][]rune{} m := [][]rune{}
parts := []partNumber{} parts := []partNumber{}
symbols := make(map[int]map[int]symbol) symbols := make(symbolTab)
symbolList := []*symbol{}
for scan.Scan() { for scan.Scan() {
text := scan.Text() text := scan.Text()
@ -77,10 +84,12 @@ func main() {
if a != '.' { if a != '.' {
cols, ok := symbols[row] cols, ok := symbols[row]
if !ok { if !ok {
cols = make(map[int]symbol) cols = make(map[int]*symbol)
} }
cols[col] = symbol{row: row, col: col} s := &symbol{row: row, col: col, symbol: a}
cols[col] = s
symbols[row] = cols symbols[row] = cols
symbolList = append(symbolList, s)
} }
} }
if v, err := strconv.Atoi(string(slice)); err == nil { if v, err := strconv.Atoi(string(slice)); err == nil {
@ -91,15 +100,24 @@ func main() {
sum := 0 sum := 0
for i, p := range parts { for i, p := range parts {
ok :=scanSymbol(symbols, p) ok := scanSymbol(symbols, p)
parts[i].hasSymbol = ok parts[i].hasSymbol = ok
if ok { if ok {
sum += p.number sum += p.number
} }
} }
sumGears := 0
for _, s := range symbolList {
if s.symbol == '*' && len(s.adjacentParts) == 2 {
sumGears += s.adjacentParts[0].number * s.adjacentParts[1].number
}
}
fmt.Println(m) fmt.Println(m)
fmt.Println(parts) fmt.Println(parts)
fmt.Println(symbols) fmt.Println(symbols)
fmt.Println(symbolList)
fmt.Println(sum) fmt.Println(sum)
fmt.Println(sumGears)
} }