assign comparator functions to variables for readability

This commit is contained in:
onyx-and-iris 2024-12-02 22:28:14 +00:00
parent 6e06bd5b16
commit d321e6c977
6 changed files with 22 additions and 18 deletions

View File

@ -2,7 +2,7 @@ package one
import "math"
func checkWithComparator(nums []int, comparator comparator) bool {
func check(nums []int, comparator comparator) bool {
for i := 1; i < len(nums); i++ {
if comparator(nums, i) || !isSafe(nums[i-1], nums[i]) {
return false

View File

@ -31,7 +31,7 @@ func TestCheckIncreased(t *testing.T) {
for name, tc := range tt {
t.Run(name, func(t *testing.T) {
got := checkWithComparator(tc.input, cmp)
got := check(tc.input, cmp)
assert.Equal(t, tc.want, got)
})
}
@ -62,7 +62,7 @@ func TestCheckDecreased(t *testing.T) {
for name, tc := range tt {
t.Run(name, func(t *testing.T) {
got := checkWithComparator(tc.input, cmp)
got := check(tc.input, cmp)
assert.Equal(t, tc.want, got)
})
}

View File

@ -23,18 +23,18 @@ func parseLines(r io.Reader) int {
scanner := bufio.NewScanner(r)
for scanner.Scan() {
nums := util.IntsFromString(scanner.Text())
if check(checkWithComparator, nums, func(nums []int, i int) bool {
cmpIncrease := func(nums []int, i int) bool {
return nums[i-1] >= nums[i]
}) || check(checkWithComparator, nums, func(nums []int, i int) bool {
}
cmpDecrease := func(nums []int, i int) bool {
return nums[i-1] <= nums[i]
}) {
}
if check(nums, cmpIncrease) || check(nums, cmpDecrease) {
sum++
}
}
return sum
}
func check(fn func([]int, comparator) bool, nums []int, comparator comparator) bool {
return fn(nums, comparator)
}

View File

@ -5,7 +5,7 @@ import (
"slices"
)
func checkWithComparator(nums []int, comparator comparator, count int) bool {
func withComparator(nums []int, comparator comparator, count int) bool {
if count > 1 {
return false
}
@ -14,8 +14,8 @@ func checkWithComparator(nums []int, comparator comparator, count int) bool {
if comparator(nums, i) || !isSafe(nums[i-1], nums[i]) {
count++
removeFirst, removeSecond := nextLevels(nums, i)
return checkWithComparator(removeFirst, comparator, count) ||
checkWithComparator(removeSecond, comparator, count)
return withComparator(removeFirst, comparator, count) ||
withComparator(removeSecond, comparator, count)
}
}

View File

@ -36,7 +36,7 @@ func TestIncreasedWithDampener(t *testing.T) {
for name, tc := range tt {
t.Run(name, func(t *testing.T) {
got := checkWithComparator(tc.input, cmp, 0)
got := withComparator(tc.input, cmp, 0)
assert.Equal(t, tc.want, got)
})
}
@ -54,7 +54,7 @@ func TestDecreasedWithDampener(t *testing.T) {
for name, tc := range tt {
t.Run(name, func(t *testing.T) {
got := checkWithComparator(tc.input, cmp, 0)
got := withComparator(tc.input, cmp, 0)
assert.Equal(t, tc.want, got)
})
}

View File

@ -23,11 +23,15 @@ func parseLines(r io.Reader) int {
scanner := bufio.NewScanner(r)
for scanner.Scan() {
nums := util.IntsFromString(scanner.Text())
if check(checkWithComparator, nums, func(nums []int, i int) bool {
cmpIncrease := func(nums []int, i int) bool {
return nums[i-1] >= nums[i]
}) || check(checkWithComparator, nums, func(nums []int, i int) bool {
}
cmpDecrease := func(nums []int, i int) bool {
return nums[i-1] <= nums[i]
}) {
}
if check(withComparator, nums, cmpIncrease) || check(withComparator, nums, cmpDecrease) {
sum++
}
}