在Swift中将数组拆分为子数组

Question

给定一个值数组，我如何将它分成由相等元素组成的子数组？

给定这个数组

let numbers = [1, 1, 1, 3, 3, 4]

我想要这个输出

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

解决这一问题的一种可能方法是创建某种索引，以指示每个元素的出现情况。

let indexes = [1:3, 3:2, 4:1]

最后使用索引重建输出数组。

let subsequences = indexes.sort { $0.0.0 < $0.1.0 }.reduce([Int]()) { (res, elm) -> [Int] in
    return res + [Int](count: elm.1, repeatedValue: elm.0)
}

但是，使用此解决方案，我会丢失原始值。当然，在这种情况下，这不是一个大问题（一个Int值仍然存在，即使重新创建，Int），但我想将此解决方案应用于像这样更复杂的数据结构

struct Starship: Equatable {
    let name: String
    let warpSpeed: Int
}

func ==(left:Starship, right:Starship) -> Bool {
    return left.warpSpeed == right.warpSpeed
}

实际上，我正在寻找的函数是与flatte（）相反的函数

let subsequences: [[Int]] = [[1,1,1], [3, 3], [4]]
print(Array(subsequences.flatten())) // [1, 1, 1, 3, 3, 4]

我希望我已经说清楚了，如果你需要更多的细节，请告诉我。

Answer 1

假设您有一个未排序的项目数组。您将需要对初始数组进行排序，然后您将得到如下内容：[1,1,1,3,3,4]

之后，您将初始化两个数组：一个用于存储数组，另一个用于将其用作当前数组。

循环遍历初始数组并执行以下操作：

• 如果当前值与上一个值没有不同，将其推送到当前数组
• 否则将当前数组推到第一个数组，然后清空当前数组。

希望有帮助！

Answer 2

如果您可以使用CocoaPods/Carthage/Swift Package Manager/等。您可以使用类似oisdk/SwiftSequence的包，它提供了group（）方法：

numbers.lazy.group()
// should return a sequence that generates [1, 1, 1], [3, 3], [4].

或UsrNameu1/TraverSwift，它提供了groupBy：

groupBy(SequenceOf(numbers), ==)

如果不想添加外部依赖项，可以编写如下算法：

func group<S: SequenceType where S.Generator.Element: Equatable>(seq: S) -> [[S.Generator.Element]] {
    var result: [[S.Generator.Element]] = []
    var current: [S.Generator.Element] = []
    for element in seq {
        if current.isEmpty || element == current[0] {
            current.append(element)
        } else {
            result.append(current)
            current = [element]
        }
    }
    result.append(current)
    return result
}

group(numbers)
// returns [[1, 1, 1], [3, 3], [4]].

Answer 3

 // extract unique numbers using a set, then
 // map sub-arrays of the original arrays with a filter on each distinct number

 let numbers = [1, 1, 1, 3, 3, 4]

 let numberGroups = Set(numbers).map{ value in return numbers.filter{$0==value} }

 print(numberGroups)

[编辑]更改为使用Hamish建议的设置初始值设定项

[EDIT2]Swift 4在字典中添加了一个初始化器，可以更有效地执行此操作：

 let numberGroups = Array(Dictionary(grouping:numbers){$0}.values)

对于要按其属性之一分组的对象列表：

 let objectGroups = Array(Dictionary(grouping:objects){$0.property}.values)