在Swift中获取数组中所有可能的项目组合而没有重复的组

Question

我正在尝试在Array上创建扩展，我可以在不生成重复组的情况下获得数组的所有可能组合，包括无项目组合。

例如，对于此数组：

[1, 2, 3, 4]

应生成以下可能的组合：

[[], [1], [2], [3], [4], [1, 2], [1, 3], [1, 4], [2, 3], [2, 4], [3, 4], [1, 2, 3], [1, 2, 4], [1, 3, 4], [2, 3, 4], [1, 2, 3, 4]]

请注意，没有一个组重复自己，即：如果有一个组[1，2]，则没有其他组：[2，1]。

这是我能得到的最接近的结果：

public extension Array {

func allPossibleCombinations() -> [[Element]] {
    var output: [[Element]] = [[]]
    for groupSize in 1...self.count {
        for (index1, item1) in self.enumerated() {
            var group = [item1]
            for (index2, item2) in self.enumerated() {
                if group.count < groupSize {
                    if index2 > index1 {
                        group.append(item2)
                        if group.count == groupSize {
                            output.append(group)
                            group = [item1]
                            continue
                        }
                    }
                } else {
                    break
                }
            }
            if group.count == groupSize {
                output.append(group)
            }
        }
    }
    return output
}

}

但是，它缺少组大小为3的项目的可能组合（我只回来了[1, 2, 3]和[2, 3, 4]。

非常感激！

Answer 1

extension Array {
    var combinations: [[Element]] {
        if count == 0 {
            return [self]
        }
        else {
            let tail = Array(self[1..<endIndex])
            let head = self[0]

            let first = tail.combinations
            let rest = first.map { $0 + [head] }

            return first + rest
        }
    }
}

print([1, 2, 3, 4].combinations)

Answer 2

您flatMap也可以将它们组合成一行。

    extension Array {
        var combinationsWithoutRepetition: [[Element]] {
            guard !isEmpty else { return [[]] }
            return Array(self[1...]).combinationsWithoutRepetition.flatMap { [$0, [self[0]] + $0] }
        }
    }

 print([1,2,3,4].combinationsWithoutRepetition)