Swift中如何高效排序字典数组并按参数拆分？

这是个很常见的开发需求，你的思路方向是对的，但确实有更高效的方式来优化性能，尤其是当处理大规模数组的时候。先帮你梳理下现有方案的问题，再给出优化后的实现：

现有方案的问题 #

你的代码里存在几个可以优化的点：

1. 手动遍历排序后的数组来构建分组字典，这个过程的性能不如标准库内置的分组方法——标准库的Dictionary(grouping:)是经过底层优化的哈希表操作，比手动append高效得多。
2. 使用字典存储元素时，直接通过键（比如"groupID"）访问值会有哈希查找的开销，而且代码里的语法也有问题（Swift字典不能直接用.访问键，得用下标）。

优化方案1：用标准库高阶函数实现（基于字典） #

我们可以先按groupID分组，再对每个分组内的元素按value排序，这样既能得到拆分后的数组，也能快速合并出全局排序后的数组：

// 修正后的输入数组（Swift字典的正确语法）
let array: [[String: Int]] = [
    ["value":3, "groupID":1],
    ["value":5, "groupID":2],
    ["value":2, "groupID":1],
    ["value":6, "groupID":3],
    ["value":1, "groupID":2],
    ["value":9, "groupID":3]
]

// 1. 按groupID分组，得到键为groupID、值为对应元素数组的字典
var groupedDict = Dictionary(grouping: array, by: { $0["groupID"]! })

// 2. 对每个分组内的元素按value升序排序
for (groupID, var groupElements) in groupedDict {
    groupElements.sort { $0["value"]! < $1["value"]! }
    groupedDict[groupID] = groupElements
}

// 期望输出2：按groupID排序后的拆分数组列表
let separatedArrays = groupedDict.sorted(by: { $0.key < $1.key }).map { $0.value }

// 期望输出1：合并所有排序后的分组，得到全局排序数组
let sortedArray = separatedArrays.flatMap { $0 }

为什么这个方案更高效？ #

• Dictionary(grouping:)的时间复杂度是O(n)，比手动遍历构建字典的性能更优，因为它是底层优化的哈希表实现。
• 分组后再对每组排序，总时间复杂度和全局排序的O(n log n)相当，但实际执行中，分组排序的缓存局部性更好，在处理大数据量时会有明显的速度优势。

优化方案2：用自定义结构体代替字典（性能更进一步） #

如果数组规模很大，推荐用自定义结构体来存储元素——结构体的属性访问是直接内存访问，比字典的键哈希查找快得多：

// 定义结构体存储元素，避免字典的键查找开销
struct DataItem {
    let value: Int
    let groupID: Int
}

// 输入数组
let items: [DataItem] = [
    .init(value: 3, groupID: 1),
    .init(value: 5, groupID: 2),
    .init(value: 2, groupID: 1),
    .init(value: 6, groupID: 3),
    .init(value: 1, groupID: 2),
    .init(value: 9, groupID: 3)
]

// 分组+排序
var groupedItems = Dictionary(grouping: items, by: { $0.groupID })
for (groupID, var group) in groupedItems {
    group.sort { $0.value < $1.value }
    groupedItems[groupID] = group
}

// 得到拆分后的数组
let separatedItemArrays = groupedItems.sorted(by: { $0.key < $1.key }).map { $0.value }
// 得到全局排序数组
let sortedItemArray = separatedItemArrays.flatMap { $0 }

这个方案在处理十万级甚至百万级元素时，性能会比字典版本提升不少，因为彻底避免了字典键的哈希计算和查找开销。

总结优化要点 #

• 优先使用Swift标准库的Dictionary(grouping:)进行分组，代替手动遍历构建分组结构。
• 对于频繁访问的结构化数据，用自定义结构体/类代替字典，减少键查找的性能开销。
• 先分组再对每组排序，逻辑清晰且性能表现更稳定。

内容的提问来源于stack exchange，提问作者riceman89