Question

给定x个数组，每个数组可能有不同数量的元素，如何迭代所有组合，我从每个数组中选择一个项目？

示例：

[   ]   [   ]   [   ]
 foo     cat      1
 bar     dog      2
 baz              3
                  4

返回

[foo]   [cat]   [ 1 ]
[foo]   [cat]   [ 2 ]
  ...
[baz]   [dog]   [ 4 ]

我在Perl中这样做，顺便说一下。

Answer 1

我的Set::CrossProduct模块完全符合您的要求。请注意，您并不是在寻找排列，这是排列中元素的排序。您正在寻找交叉产品，它是来自不同集合的元素的组合。

我的模块为您提供了一个迭代器，因此您不会在内存中创建它。只有在需要时才创建新元组。

use Set::Crossproduct;

my $iterator = Set::CrossProduct->new(
    [
        [qw( foo bar baz )],
        [qw( cat dog     )],
        [qw( 1 2 3 4     )],
    ]
    );

while( my $tuple = $iterator->get ) {
    say join ' ', $tuple->@*;
    }

Answer 2

任意数量列表的简单递归解决方案：

sub permute {
  my ($first_list, @remain) = @_;

  unless (defined($first_list)) {
    return []; # only possibility is the null set
  }

  my @accum;
  for my $elem (@$first_list) {
    push @accum, (map { [$elem, @$_] } permute(@remain));
  }

  return @accum;
}

任意数量的列表的一个不那么简单的非递归解决方案：

sub make_generator {
  my @lists = reverse @_;

  my @state = map { 0 } @lists;

  return sub {
    my $i = 0;

    return undef unless defined $state[0];

    while ($i < @lists) {
      $state[$i]++;
      last if $state[$i] < scalar @{$lists[$i]};
      $state[$i] = 0;
      $i++;
    }

    if ($i >= @state) {
      ## Sabotage things so we don't produce any more values
      $state[0] = undef;
      return undef;
    }

    my @out;
    for (0..$#state) {
      push @out, $lists[$_][$state[$_]];
    }

    return [reverse @out];
  };
}

my $gen = make_generator([qw/foo bar baz/], [qw/cat dog/], [1..4]);
while ($_ = $gen->()) {
  print join(", ", @$_), "\n";
}

Answer 3

可以在http://www.perlmonks.org/?node_id=7366找到用于执行笛卡尔积的递归和更流畅的Perl示例（带注释和文档）

示例：

sub cartesian {
    my @C = map { [ $_ ] } @{ shift @_ };

    foreach (@_) {
        my @A = @$_;

        @C = map { my $n = $_; map { [ $n, @$_ ] } @C } @A;
    }

    return @C;
}

Answer 4

我首先想到的一种方法是使用一对for循环而不是递归。

查找排列总数
从0循环到total_permutations-1
观察一下，通过取循环索引模数数组中的元素数，可以得到每个排列

示例：

给定A [3]，B [2]，C [3]，

for (index = 0..totalpermutations) {
    print A[index % 3];
    print B[(index / 3) % 2];
    print C[(index / 6) % 3];
}

当然，for循环可以代替循环[A B C ...]，并且可以记忆一小部分。当然，递归更简洁，但这对于递归受到堆栈大小严重限制的语言可能很有用。

Answer 5

您可以使用嵌套循环。

for my $e1 (qw( foo bar baz )) {
for my $e2 (qw( cat dog )) {
for my $e3 (qw( 1 2 3 4 )) {
   my @choice = ($e1, $e2, $e3); 
   ...
}}}

当您需要任意数量的嵌套循环时，可以使用Algorithm::Loops的NestedLoops。

use Algorithm::Loops qw( NestedLoops );

my @lists = (
   [qw( foo bar baz )],
   [qw( cat dog )],
   [qw( 1 2 3 4 )],
);

my $iter = NestedLoops(\@lists);
while ( my @choice = $iter->() ) {
   ...
}

在Perl中，我如何迭代多个集合的笛卡尔积？

5 个答案: