Intern
""" Места у биоскопу су распоређена у једном реду. Нови гледалац који је управо стигао бира место, да седи што даље од осталих гледалаца у реду. То јест, растојање од тог места, до најближе гледаоца, треба да буде максимално. Гарантовано је да увек постоје слободна места и да већ седи бар један гледалац. Напишите функцију која, датом редом места (массив нула и јединица), враћа растојање (број интервала између седишта) од изабраного места до најближег гледаоца. [1, 0, 0, 0, 1] -> 2 [1, 0, 1, 0, 0, 1, 0, 0, 1] -> 2 [1, 0, 1, 0] -> 1 """
sobes.tech АИ
Одговор од АИ
go package main
import ( "fmt" "math" )
func maxDistance(seats []int) int { maxDist := 0 lastOccupied := -1
for i, seat := range seats {
if seat == 1 {
if lastOccupied == -1 {
// расстояние от начала ряда до первого занятого места
maxDist = i
} else {
// расстояние между двумя занятыми местами
midDist := (i - lastOccupied) / 2
if midDist > maxDist {
maxDist = midDist
}
}
lastOccupied = i
}
}
// расстояние от последнего занятого места до конца ряда
tailDist := len(seats) - 1 - lastOccupied
if tailDist > maxDist {
maxDist = tailDist
}
return maxDist
}
func main() { fmt.Println(maxDistance([]int{1, 0, 0, 0, 1})) // 2 fmt.Println(maxDistance([]int{1, 0, 1, 0, 0, 1, 0, 0, 1})) // 2 fmt.Println(maxDistance([]int{1, 0, 1, 0})) // 1 }
Объяснение:
- Идём по ряду, запоминая индекс последнего занятого места.
- Для каждого занятого места считаем расстояние до предыдущего и берём половину (максимально удалённое место между ними).
- Также учитываем расстояния от начала ряда до первого занятого места и от последнего занятого места до конца ряда.
- Возвращаем максимальное из этих расстояний.