4

我需要从另一个包含所有可能组合的列表中创建一个列表。在研究可能的解决方案时,我发现了许多有趣的方法,但似乎都是根据提供的记录数生成结果。我需要将组合增加到最大阈值。

即考虑以下数组

1,2,3,4,5

我需要结果看起来类似于(在此示例中阈值为 3)

1
1,2
1,2,3
1,2,4
1,2,5
1,3,4
2,3,5... etc

实际上,数据将是 IEnumerable。我使用了一个简单的 int[] 来说明所需的结果。

4

3 回答 3

5

我的解决方案使用简单的递归算法来创建组合:

  • 当我们遍历序列时,我们可以立即返回一个只保存当前值的序列。我编写了一个简单的扩展方法来为单个项目创建一个 IEnumerable。

  • 接下来,我们递归地生成剩余元素的所有组合,并将阈值减少 1,并将它们中的每一个与当前值组合。

我假设元素不应重复(即 { 1, 1 } 或 { 1, 2, 1 } 是不允许的)。如果您想允许重复的元素,您可以删除变量并在递归调用中将其remaining替换为.valuesGetCombinations

请注意yield关键字的使用。这意味着代码使用延迟执行。在实际枚举结果之前,无需存储任何中间结果。

public static IEnumerable<IEnumerable<T>> GetCombinations<T>(IEnumerable<T> values, int threshold)
{
    var remaining = values;

    foreach (T value in values)
    {
        yield return value.Yield();

        if (threshold < 2)
        {
            continue;
        }

        remaining = remaining.Skip(1);

        foreach (var combination in GetCombinations(remaining, threshold - 1))
        {
            yield return value.Yield().Concat(combination);
        }
    }
}

public static IEnumerable<T> Yield<T>(this T item)
{
    yield return item;
}

对于整数数组 { 1, 2, 3, 4, 5 },输出为:

1
1, 2
1, 2, 3
1, 2, 4
1, 2, 5
1, 3
1, 3, 4
1, 3, 5
1, 4
1, 4, 5
1, 5
2
2, 3
2, 3, 4
2, 3, 5
2, 4
2, 4, 5
2, 5
3
3, 4
3, 4, 5
3, 5
4
4, 5
5
于 2013-10-04T23:26:27.887 回答
1

假设您已经有一个解决方案来查找特定计数的组合(您已经说过您有),让我们说一下签名:

public static IEnumerable<IEnumerable<T>> Combinations<T>(
    IList<T> source, int count)

然后,您可以通过调用 N 次轻松获得所有计数的组合:

public static IEnumerable<IEnumerable<T>> Combinations<T>(IList<T> source)
{
    return Enumerable.Range(0, source.Count)
            .SelectMany(i => Combinations(source, i));
}
于 2013-10-04T18:51:17.867 回答
1

这是一种解决方案:

public static IEnumerable<T[]> Combinations<T>(IEnumerable<T> items, int threshold)
{
    var comboBuilder = new List<T>();
    foreach (var combo in EnumerateCombos(items, comboBuilder, 0, threshold))
    {
        yield return combo;
    }
}
private static IEnumerable<T[]> EnumerateCombos<T>(IEnumerable<T> items, List<T> currentCombo, 
                                                   int startIndex, int threshold)
{
    if (currentCombo.Count >= threshold) { yield break; }

    for (int i = startIndex; i < items.Count(); i++)
    {
        //Skip past the items we've already gone through in the current combo:
        var item = items.Skip(i).First();

        //Create a new combination with the next available item:
        currentCombo.Add(item);
        yield return currentCombo.ToArray();

        //Repeat the process with the rest of the remaining items:
        foreach (var combo in EnumerateCombos(items, currentCombo, i + 1, threshold))
        {
            yield return combo;
        }

        //Recursion cleanup:
        currentCombo.RemoveAt(currentCombo.Count - 1);
    }
}
于 2013-10-04T20:24:01.820 回答