Получить все индексы значения nil в массиве

У меня есть массив N элементы и этот массив содержит либо 0, 1 или же nil, Я хочу получить все индексы, на которых nil присутствует или сортировать массив так, чтобы все nil приходит первым

Я ищу эффективный способ, потому что размер массива может быть очень большим.

Вот мой код

array_of data # array with lots of 1, 0 and nil
temp = []
array_of_data.each_with_index {|a,i| (array_of_data[i] ? true : temp << i )}

1 ответ

Решение

Альтернативы для вас для сравнения:

# Indices of non-nil values
res = ary.map.with_index{ |v,i| i if v }.compact
res = [].tap{ |r| ary.each.with_index{ |v,i| r<<i if v } }
ary.map!.with_index{ |v,i| v && i }.compact

# Sorting the array so that nil comes first (possibly re-ordering the others)
res = ary.sort_by{ |v| v ? 1 : -1 }
ary.sort_by!{ |v| v ? 1 : -1 }

# Sorting the array so that nil comes first, order of others unchanged
res = ary.sort_by.with_index{ |v,i| v ? i : -1 }
ary.sort_by!.with_index{ |v,i| v ? i : -1 }
Другие вопросы по тегам