Ruby:Mutating方法参数(整数与数组)

时间:2017-08-03 03:15:49

标签: ruby pass-by-reference pass-by-value

在下面的代码中,我的初衷是将计数器变量传递给一个辅助方法,该方法以递归方式运行,沿途更新它,然后在辅助函数运行完毕后返回它。

计数器返回零,我正试图弄清楚为什么会这样。我知道这种方法在你传入一个数组时有效,我相信这是因为每个对数组的引用都指向内存中的同一个对象,例如当你使用铲运算符时,你就是在改变那个对象。

然而,在这种情况下,这不是正在发生的事情,并且计数器变量被重新分配给内存中的新值(无论结果整数指向何处)。

这是我的问题 - >这里的问题是计数器变量是否正确地被重新分配,但仅在其定义的范围内?因此,返回'counter + = 1'不会保留对计数器变量的更改,因为在该函数的范围内(并且仅在函数的范围内),计数器变量会被重新分配?难以彻底理解这一点。以下是感兴趣的人的问题陈述:

给定一个包含所有正数但没有重复数的整数数组,找到加起来为正整数目标的可能组合数。

Example:

nums = [1, 2, 3]
target = 4

The possible combination ways are:
(1, 1, 1, 1)
(1, 1, 2)
(1, 2, 1)
(1, 3)
(2, 1, 1)
(2, 2)
(3, 1)

Note that different sequences are counted as different combinations.

Therefore the output is 7.

这是我的代码:

# @param {Integer[]} nums
# @param {Integer} target
# @return {Integer}
def combination_sum4(nums, target)
    nums.sort! # n log n 
    generate_combinations(nums, target, counter = 0)
    counter
end

def generate_combinations(nums, target, counter)
    return counter += 1 if target == 0
    (1...nums.length).each do |num|
        break if num > target
        generate_combinations(nums, target - num, counter)
    end 
    counter
end 

2 个答案:

答案 0 :(得分:0)

这是一个范围问题。您传入counter的本地变量generate_combinations与其中的counter不同。方法中定义的局部变量的范围限定为该方法。当您将它们传递给另一个方法时,您传递的是变量的值,而不是变量本身。在combination_sum4内,counter永远不会更改其原始值0

解决方案

1)使用实例变量@counter,只要它们在同一个上下文中定义,或者在同一个类或模块上,就可以由这些方法共享。如果您稍后需要在后续方法调用期间访问此相同值,这将非常有用。

2)不要在counter结束时返回combination_sum4。将generate_combinations保留为方法的最后一行。无论它返回什么,也将由combination_sum4返回。

def combination_sum4(nums, target)
  nums.sort! # n log n 
  generate_combinations(nums, target)
end

def generate_combinations(nums, target, counter = 0) # `counter` is now an optional argument with a default value of `0`
  return counter + 1 if target.zero?
  nums.each do |num| # use inclusive range (..) rather than exclusive range (...)
    break if num > target
    counter = generate_combinations(nums, target - num, counter) # reassign `counter` to the return value of the recursion
  end 
  counter
end

答案 1 :(得分:0)

@ m-simon-borg已经完美地解释了你的问题:你有一个范围问题。

但是,我会建议使用inject(@see:https://ruby-doc.org/core-2.1.0/Enumerable.html#method-i-reduce)来实现这一目标:

def combination_sum4(nums, target)
  generate_combinations(nums.sort, target, 0)
end

def generate_combinations(nums, target, counter)
  return counter += 1 if target == 0

  (1..nums.length).inject(counter) do |counter, num|
    num > target ?
      counter : 
      generate_combinations(nums, target - num, counter)
  end
end

nums = [1, 2, 3]
target = 4
combination_sum4(nums, target) # 7