Выдает ошибку при решении задачи на Golang

Question

Дан неотсортированный слайс целых чисел. Напишите функцию UnderLimit(nums []int, limit int, n int) ([]int, error), которая будет возвращать первые n (либо меньше, если остальные не подходят) элементов, которые меньше limit. В случае ошибки функция должна вернуть nil и описание ошибки. Пишу что надо -- выдает ошибку

Answer 1

Попробуйте такое решение:

func UnderLimit(nums []int, limit int, n int) ([]int, error) {
    if n < 0 {
        return nil, fmt.Errorf("n cannot be negative")
    }
    if nums == nil {
        return nil, fmt.Errorf("nums cannot be nil")
    }
    if n == 0{
        return []int{}, nil
    }

    result := make([]int, 0, n)
    count := 0
    for _, num := range nums {
        if count == n {
            return result, nil
        }
        if num < limit {
            result = append(result, num)
            count++
        }
    }
    return result, nil
}

БЛОГ НА HUSL

Выдает ошибку при решении задачи на Golang

Ответы (1 шт):