feat: solve part one using binary search on sorted range
This commit is contained in:
79
internal/2025/DayFive/code.go
Normal file
79
internal/2025/DayFive/code.go
Normal file
@@ -0,0 +1,79 @@
|
||||
package dayfive
|
||||
|
||||
import (
|
||||
"os"
|
||||
"slices"
|
||||
"sort"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"advent-of-code/internal/registry"
|
||||
)
|
||||
|
||||
type freshRange struct {
|
||||
start int
|
||||
end int
|
||||
}
|
||||
|
||||
func init() {
|
||||
registry.Register("2025D5", ParseInput, PartOne, PartTwo)
|
||||
}
|
||||
|
||||
func ParseInput(filepath string) []string {
|
||||
content, _ := os.ReadFile(filepath)
|
||||
return strings.Split(string(content), "\n")
|
||||
}
|
||||
|
||||
func PartOne(input []string) int {
|
||||
var freshRanges []freshRange
|
||||
var ingredientIDs []int
|
||||
separatorFound := false
|
||||
|
||||
for _, line := range input {
|
||||
if line == "" {
|
||||
separatorFound = true
|
||||
continue
|
||||
}
|
||||
|
||||
if !separatorFound {
|
||||
startStr, endStr, _ := strings.Cut(line, "-")
|
||||
start, _ := strconv.Atoi(startStr)
|
||||
end, _ := strconv.Atoi(endStr)
|
||||
freshRanges = append(freshRanges, freshRange{start: start, end: end})
|
||||
} else {
|
||||
id, _ := strconv.Atoi(line)
|
||||
ingredientIDs = append(ingredientIDs, id)
|
||||
}
|
||||
}
|
||||
|
||||
slices.SortFunc(freshRanges, func(a, b freshRange) int {
|
||||
switch {
|
||||
case a.start < b.start:
|
||||
return -1
|
||||
case a.start > b.start:
|
||||
return 1
|
||||
default:
|
||||
return 0
|
||||
}
|
||||
})
|
||||
|
||||
freshCount := 0
|
||||
for _, id := range ingredientIDs {
|
||||
idx := sort.Search(len(freshRanges), func(idx int) bool {
|
||||
return freshRanges[idx].start > id
|
||||
})
|
||||
|
||||
for i := idx - 1; i >= 0 && freshRanges[i].start <= id; i-- {
|
||||
if id <= freshRanges[i].end {
|
||||
freshCount++
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return freshCount
|
||||
}
|
||||
|
||||
func PartTwo(input []string) int {
|
||||
return 0
|
||||
}
|
||||
Reference in New Issue
Block a user