Middle
import "fmt" func maxDistants(arr []int) int { maxDist := 0 lastOne := -1 for i := 0; i < len(arr); i++ { if arr[i] == 1 { if lastOne == -1 { maxDist = i } else { dist := (i - lastOne) / 2 if dist > maxDist { maxDist = dist } } lastOne = i } } if arr[len(arr)-1] == 0 { dist := len(arr) - 1 - lastOne if dist > maxDist { maxDist = dist } } return maxDist } func main() { fmt.Println(maxDistants([]int{1, 0, 0, 0, 1})) }
sobes.tech AI
Odpoveď od AI
Funkcia maxDistants v Go vypočíta maximálnu vzdialenosť k najbližšiemu prvku s hodnotou 1 v poli pozostávajúcom z 0 a 1. Myšlienka je nájsť pozíciu, kde je možné umiestniť 1 tak, aby vzdialenosť k najbližšiemu 1 bola maximálna.
Vysvetlenie logiky:
- Premenná
lastOneuchováva index posledného nájdeného 1. - Pri prechádzaní po poli, ak narazíme na 1:
- Ak je to prvý 1, maximálna vzdialenosť je index tohto 1 (vzdialenosť od začiatku).
- Inak vypočítame polovicu vzdialenosti medzi aktuálnym a predchádzajúcim 1 (pretože ho môžeme umiestniť do stredu).
- Po prechode kontrolujeme, či napravo od posledného 1 sú nuly, a aktualizujeme maximálnu vzdialenosť podľa potreby.
Príklad použitia:
package main
import "fmt"
func maxDistants(arr []int) int {
maxDist := 0
lastOne := -1
for i := 0; i < len(arr); i++ {
if arr[i] == 1 {
if lastOne == -1 {
maxDist = i
} else {
dist := (i - lastOne) / 2
if dist > maxDist {
maxDist = dist
}
}
lastOne = i
}
}
if arr[len(arr)-1] == 0 {
dist := len(arr) - 1 - lastOne
if dist > maxDist {
maxDist = dist
}
}
return maxDist
}
func main() {
fmt.Println(maxDistants([]int{1, 0, 0, 0, 1})) // 2
}
Tento kód správne funguje na nájdenie maximálnej vzdialenosti k najbližšiemu 1, pričom zohľadňuje okraje poľa.