Swift中如何使用map实现[String: [String]]字典转指定规则字符串数组

实现方案 #

这种每个键值对处理后生成一组子数组、最终要合并为一维数组的场景，用 flatMap 替代手动嵌套循环+可变数组追加的逻辑最合适，逻辑和你原来的手写实现完全等价：

1. 逐个遍历字典的键值对
2. 当值数组为空时，返回仅包含对应key的单元素数组
3. 当值数组非空时，对数组内的每个元素和对应key做字符串拼接，返回所有拼接结果组成的数组
4. flatMap 会自动把每一步返回的子数组合并、拍平成最终的一维结果数组，不需要手动维护可变数组。

核心实现代码 #

let messages: [String: [String]] = ["A": ["1","2"], "b": [], "c": ["5", "6"]]

let result = messages.flatMap { key, values in
    values.isEmpty ? [key] : values.map { key + $0 }
}

逻辑说明 #

• 代码全程不需要声明可变的var result，也不需要手写多层for循环、append操作，消除了冗余的样板代码。
• 内层的values.map负责遍历非空的value集合，完成key与单个value的字符串拼接。
• 外层的flatMap自动完成子数组合并：如果这里用普通的map遍历字典，会得到[[String]]类型的二维数组，还需要额外做一次展平操作，而flatMap自带拍平逻辑，刚好匹配需求，最终生成的元素集合和你手写循环的输出完全一致。

顺序补充说明 #

Swift的Dictionary是无序哈希结构，直接遍历得到的元素顺序不固定。如果需要固定输出顺序（比如按key的字典序排列），可以在遍历前先对键值对做排序处理：

// 按key的升序排序后再生成结果，输出顺序固定
let result = messages
    .sorted(by: { $0.key < $1.key })
    .flatMap { key, values in
        values.isEmpty ? [key] : values.map { key + $0 }
    }

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