LeetCode/go/get-equal-substrings-within-budget.go
Matej Focko d2ef757754
go: allow testing
Signed-off-by: Matej Focko <me@mfocko.xyz>
2024-06-16 11:44:08 +02:00

29 lines
416 B
Go

package main
func equalSubstring(s string, t string, maxCost int) int {
abs := func(x int) int {
return max(x, -x)
}
getCost := func(i int) int {
return abs(int(s[i]) - int(t[i]))
}
mostOptimal := 0
begin := 0
cost := 0
for i, _ := range t {
cost += getCost(i)
for cost > maxCost {
cost -= getCost(begin)
begin++
}
mostOptimal = max(mostOptimal, i-begin+1)
}
return mostOptimal
}