Поиск через матрицу с Руби

Question

У меня есть матрица вроде этого:

0 1 0 0 1 0 
1 0 1 0 1 0 
0 1 0 1 0 0 
0 0 1 0 1 1 
1 1 0 1 0 0 
0 0 0 1 0 0 

Как я могу определить матрицу в Ruby, а затем искать через него?

Я хотел бы написать программу, которая просматривает все строки и возвращает строки с наибольшей суммой «1».

Answer 1

Быстрый способ определить матрицу в Ruby:

Array.new 6, Array.new(6, 0) 
# => [ 
     [0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0] 
    ] 

Приведенный выше код инициализирует массив с 6 и по умолчанию их значения 2-го аргумента, который является другой массив с 6 и значениями 0 по умолчанию.

В других более императивных языках вы должны использовать вложенные циклы:

matrix = [] 
for x in [0,1,2,3,4,5] 
    for y in [0,1,2,3,4,5] 
    matrix[x] ||= [] # create the row as an empty array 
    matrix[x] << y # push the y value to it 
    end 
end 
# matrix is now: 
# => [ 
     [0, 1, 2, 3, 4, 5], 
     [0, 1, 2, 3, 4, 5], 
     [0, 1, 2, 3, 4, 5], 
     [0, 1, 2, 3, 4, 5], 
     [0, 1, 2, 3, 4, 5], 
     [0, 1, 2, 3, 4, 5] 
    ] 

Для поиска через матрицу и найти строку с наибольшей суммой:

greatest_sum_row_index = 0 
greatest_sum = 0 

matrix.each_with_index do |row, i| 
    # row.inject(:+) is a shortcut to adding each int in the array and returning the sum 
    sum = row.inject(:+) 
    if sum > greatest_sum 
    greatest_sum = sum 
    greatest_sum_row_index = i 
    end 
end 

# highest_row is now the index of the greatest sum row 
matrix[greatest_sum_row_index] # returns the row with the greatest sum 

Answer 2

Если массив был большим и требование к памяти было беспокойство, вы могли бы сделать что-то вроде этого:

def rows_with_most_ones(arr) 
    arr.each_with_index.with_object([-1, nil]) do |(row,i),best| 
    tot = row.count(1) 
    case tot<=>best.first 
    when 1 then best.replace([tot, [i]]) 
    when 0 then best.last << i 
    end 
    end 
end 

arr = [[0, 1, 0, 0, 1, 0], 
     [1, 0, 1, 0, 1, 0], 
     [0, 1, 0, 1, 0, 0], 
     [0, 0, 1, 0, 1, 1], 
     [1, 1, 0, 1, 0, 0], 
     [0, 0, 0, 1, 0, 0]] 

rows_with_most_ones(arr) 
    #=> [3, [1, 3, 4]]