ruby - 了解Ruby的repeated_permutation方法的源代码

Question

我一直在用 Ruby 构建一个智能 Mastermind 游戏。在我的游戏中，如果您选择让计算机扮演代码破译者的角色，计算机就会对代码编写者的代码进行有根据的猜测。

作为我的算法的一部分，计算机首先查看所有可能代码的整个列表。

例如，有6种颜色可供选择(红、橙、蓝、绿、紫、黄)，而代码由4种颜色组成(允许重复)，那么要查看所有可能的代码，可以这样做:

valid_colors = %w(red orange blue green purple yellow)
all_possible_codes = valid_colors.repeated_permutation(4).to_a

并且all_possible_codes将是一个数组，其中填充了代表每个可能的代码的数组。然后，当计算机从每个猜测中获得反馈时，就会从该列表中消除代码。

但是，我接下来要做的事情需要我使用 JRuby 1.6.6，它使用 Ruby 1.8.7，而 Ruby 1.8.7 没有 repeated_permutation 方法。我需要编写自己的具有相同功能的方法。

所以我找到了这里的源代码:http://www.ruby-doc.org/core-1.9.3/Array.html#method-i-repeated_permutation

不幸的是，我不明白他们在做什么，也不明白我如何通过编写自己的方法来解决这个问题。我对编程相当陌生，还无法弄清楚这一点。任何有助于理解源代码的帮助将不胜感激!

Answer 1

您链接的代码调用 rpermute0 来完成大部分工作，Array.c 中 rpermute0 的源代码是

static void
rpermute0(long n, long r, long *p, long index, VALUE values)
{
    long i, j;
    for (i = 0; i < n; i++) {
    p[index] = i;
    if (index < r-1) {              /* if not done yet */
        rpermute0(n, r, p, index+1, values); /* recurse */
    }
    else {
        /* We have a complete permutation of array indexes */
        /* Build a ruby array of the corresponding values */
        /* And yield it to the associated block */
        VALUE result = rb_ary_new2(r);
        VALUE *result_array = RARRAY_PTR(result);
        const VALUE *values_array = RARRAY_PTR(values);

        for (j = 0; j < r; j++) result_array[j] = values_array[p[j]];
        ARY_SET_LEN(result, r);
        rb_yield(result);
        if (RBASIC(values)->klass) {
        rb_raise(rb_eRuntimeError, "repeated permute reentered");
        }
    }
    }
}

基本上是一种蛮力，从 0 开始，每次迭代返回一个排列。 ruby 版本类似于

require 'pp'

def rpermute(numRepeat, pArray, index, originArray)
  0.upto(originArray.length-1) do |i|
    pArray[index] = i
    if index < numRepeat-1
      rpermute(numRepeat, pArray, index+1, originArray)
    else
      result = Array.new
      0.upto(numRepeat-1) do |j|
        result[j] = originArray[pArray[j]]
      end
     pp result
    end
  end
end

originArray1 = [1,2,3,4,5]
originArray2 = ['a','b','c','d','e']
pArray = []

rpermute(4, pArray, 0, originArray1)
rpermute(4, pArray, 0, originArray2)

我测试了上面的代码，它打印出了长度为 4 的所有排列，你可能想将它们放入数组中。