一个leetcode的Ruby二维数组问题

Question

请问下面的Ruby代码最后一个测试返回的data数组为什么值都是4呢？

# @param {Integer[][]} mat
# @param {Integer} r
# @param {Integer} c
# @return {Integer[][]}
def matrix_reshape(mat, r, c)
  return mat if r * c != mat.count * mat[0].count
  data = Array.new(r, Array.new(c, 0))
  ii = 0
  jj = 0
  for i in 0...(mat.count)
    for j in 0...(mat[0].count)
      data[ii][jj] = mat[i][j]
      puts "#{ii}, #{jj} -> #{data[ii][jj]}"
      jj += 1
      if jj >= c 
        jj = 0
        ii += 1
      end
    end
  end
  
  for ii in 0...data.count
    for jj in 0...data[ii].count
      puts "#{ii}, #{jj} -> #{data[ii][jj]}"
    end
  end
  return data
end

for i, r in {[[[1,2],[3,4]], 1, 4] => [[1,2,3,4]], [[[1,2],[3,4]], 2, 4] => [[1,2],[3,4]], [[[1,2],[3,4]], 4, 1] => [[1],[2],[3],[4]]}
  a = matrix_reshape(i[0], i[1], i[2])
  puts "input: #{i}, output: #{a}, expected: #{r} -> #{a == r ? 'pass' : 'fail'}"
end

Answer 1

引自强大而免费的微信小程序：皆我百晓生。

---

问题出在这一行代码 data = Array.new(r, Array.new(c, 0)) 中。在 Ruby 中，Array.new 的行为与预期可能不同，如果在 Array.new 中使用可变对象（如数组），则该对象会被共享，并在数组的每个位置上都是同一个对象的引用。

因此，在你的代码中，data 数组的每个元素都是指向同一个内存引用的数组 [0, 0, 0, 0]。当你更新一个位置的值时，其他位置也会随之改变。

要解决这个问题，你可以使用一个循环来创建 data 数组，并为每个位置都创建一个新的数组对象。以下是修改后的代码：

def matrix_reshape(mat, r, c)
  return mat if r * c != mat.count * mat[0].count
  data = Array.new(r) { Array.new(c, 0) }  # 创建一个包含 r 个新数组的数组
  ii = 0
  jj = 0
  for i in 0...(mat.count)
    for j in 0...(mat[0].count)
      data[ii][jj] = mat[i][j]
      jj += 1
      if jj >= c 
        jj = 0
        ii += 1
      end
    end
  end
  
  return data
end

通过使用 Array.new(r) { Array.new(c, 0) }，我们可以确保每个位置都得到一个新的数组对象。现在，测试应该返回预期的结果。