将整数列表 (IEnumerable<int>) 转换为特定长度的元组列表 (IEnumerable<IEnumerable<int>>)

Question

我尝试解决下一个练习：

输入：整数列表，其中count >= 1;和某个正整数k。

输出：这些整数的所有可能元组，长度为k。

例如：

输入：{1, 2}; k = 4 输出：

{
 {1, 1, 1, 1},
 {1, 1, 1, 2},
 {1, 1, 2, 1},
 {1, 1, 2, 2},
 {1, 2, 1, 1},
 {1, 2, 1, 2},
 {1, 2, 2, 1},
 {1, 2, 2, 2},
 {2, 1, 1, 1},
 {2, 1, 1, 2},
 {2, 1, 2, 1},
 {2, 1, 2, 2},
 {2, 2, 1, 1},
 {2, 2, 1, 2},
 {2, 2, 2, 1},
 {2, 2, 2, 2}
}

我试图创建一个包含k个输入列表副本的数组，然后使用Combinations：

public static IEnumerable<IEnumerable<T>> Combinations<T>(
  this IEnumerable<T> elements, 
  int k)
{
    return k == 0 
        ? new[] { new T[0] } 
        : elements.SelectMany((e, i) => elements
              .Skip(i + 1)
              .Combinations(k - 1)
              .Select(c => (new[] { e }).Concat(c)));
}

但是当 k > 9 时，这需要太长时间了。是否有一种算法能在短时间内解决这个问题？

- Andrii Khomiak

---

你将会拥有Math.Power(elements.Count(), k)个项目; 在你的情况下，仅有2**9 == 512个项目。 - Dmitry Bychenko

Answer 1

让我们放弃递归，并拥有512个项目：

代码:

//TODO: you may want to declare it as IEnumerable<T[]> Combinations<T> 
public static IEnumerable<IEnumerable<T>> Combinations<T>(
  this IEnumerable<T> elements, int k) {

  if (null == elements)
    throw new ArgumentNullException(nameof(elements));
  else if (k < 0)
    throw new ArgumentOutOfRangeException(nameof(k));

  T[] alphabet = elements.ToArray();

  // Special cases
  if (alphabet.Length <= 0)
    yield break;
  else if (k == 0)
    yield break;

  int[] indexes = new int[k];

  do {
    yield return indexes
      .Select(i => alphabet[i])
      .ToArray();

    for (int i = indexes.Length - 1; i >= 0; --i)
      if (indexes[i] >= alphabet.Length - 1)
        indexes[i] = 0;
      else {
        indexes[i] += 1;

        break;
      }
  }
  while (!indexes.All(index => index == 0));
}

演示：

string report = string.Join(Environment.NewLine, Combinations(new int[] { 1, 2}, 9)
  .Select(line => string.Join(", ", line)));

Console.Write(report);

结果：（512 条记录）

1, 1, 1, 1, 1, 1, 1, 1, 1
1, 1, 1, 1, 1, 1, 1, 1, 2
1, 1, 1, 1, 1, 1, 1, 2, 1
1, 1, 1, 1, 1, 1, 1, 2, 2
1, 1, 1, 1, 1, 1, 2, 1, 1
1, 1, 1, 1, 1, 1, 2, 1, 2
1, 1, 1, 1, 1, 1, 2, 2, 1
1, 1, 1, 1, 1, 1, 2, 2, 2
1, 1, 1, 1, 1, 2, 1, 1, 1
...
2, 2, 2, 2, 2, 2, 1, 2, 1
2, 2, 2, 2, 2, 2, 1, 2, 2
2, 2, 2, 2, 2, 2, 2, 1, 1
2, 2, 2, 2, 2, 2, 2, 1, 2
2, 2, 2, 2, 2, 2, 2, 2, 1
2, 2, 2, 2, 2, 2, 2, 2, 2

让我们生成约 2 ** 20 == 1048576 个项目（k = 20），即大小为 20 的数组超过100万个：

  Stopwatch sw = new Stopwatch();

  sw.Start();  

  int count = Combinations(new int[] { 1, 2 }, 20).Count();

  sw.Stop();

  Console.Write($"{count.ToString()} items at {sw.ElapsedMilliseconds:f0} milliseconds");

结果：

  1048576 items at 469 milliseconds

- Dmitry Bychenko