查找具有匹配键的字典值的总和答案

【问题标题】：Finding the sum of the values of a dictionary with matching keys查找具有匹配键的字典值的总和
【发布时间】：2020-08-08 18:00:16
【问题描述】：

我有一个 [String: [String: Int]] 类型的嵌套字典，如下所示：

var dict = [
    "A": ["a": 234, "b": 322],
    "B": ["c": 45, "d": 443],
    "C": ["b": 50, "e": 940, "f": 9430, "a": 53029]
]

我想把它转换成所有相同的小写键的整数加在一起形成另一个字典：

["a": 53263, "b": 382, "c": 45, "d": 50, "e": 940, "f": 9430]

我不关心密钥"A", "B", "C"。

首先，我映射dict来提取嵌套字典：

let result1 = dict.map { $1 }
// [["b": 322, "a": 234], ["c": 45, "d": 443], ["e": 940, "b": 50, "a": 53029, "f": 9430]]

我使用flatMap 来获取元组：

let result2 = dict.map { $1 }.flatMap { $0.map { $0 }}
// [(key: "c", value: 45), (key: "d", value: 443), (key: "f", value: 9430), (key: "e", value: 940), (key: "a", value: 53029), (key: "b", value: 50), (key: "a", value: 234), (key: "b", value: 322)]

最后，我尝试使用result2 并映射每个项目以添加相同键的所有值并返回[String: Int] 类型的字典：

var tempArray = [String]()
var finalDict = [String: Int]()
var count = 0
let result3 = dict.map { $1 }.flatMap { $0.map { el -> [String: Int] in
    if tempArray.contains(el.key) {
        if let newCount = finalDict[el.key] {
            count += newCount
            finalDict[el.key] = newCount
        }
    } else {
        tempArray.append(el.key)
        finalDict = [el.key: el.value]
    }

    return finalDict
}}

但是，我得到的结果是：

[["b": 322], ["a": 234], ["c": 45], ["d": 443], ["d": 443], ["d": 443], ["f": 9430], ["e": 940]]

显示多个相同的键。

【问题讨论】：

您的dict 不是您所说的 [String: [String: Int]] 类型，它似乎是 [String: [[String: Int]]]。这是正确的吗？
您完全正确，但这实际上是重新创建我的示例时的拼写错误。让我再试一次

标签： ios swift dictionary higher-order-functions

【解决方案1】：

您可以使用flatMap 创建一个元组数组，然后使用reduce(into:) 创建一个字典，其中包含键和每个键的值的总和

let result = dict.values
    .flatMap {$0}
    .reduce(into: [:]) { $0[$1.key, default: 0] += $1.value }

【讨论】：

你能解释一下$0[$1.key, default: 0]吗？
查找键 $1.key 的值，如果不存在，则插入键的值 0

【解决方案2】：

你快到了。 result2 是一个元组数组，您可以通过添加它们的值来使用它来构造具有唯一键的字典：

var dict = Dictionary(result2, uniquingKeysWith: +)

这将为您提供所需的东西（不保证订单）：

["c": 45, "a": 53263, "d": 443, "b": 372, "f": 9430, "e": 940]

有一种更简单的方法可以获取您的result2：

let result2 = dict.flatMap { $0.value }.flatMap { $0.map{ $0 } }

【讨论】：

我尝试了Dictionary(result3, uniquingKeysWith: +)，但我收到了Circular reference 错误。抱歉，我不得不稍微修改一下问题，所以 result3 现在是 result2，但元组基本相同。
@Kevvv，嗯..不确定循环引用，但如果你基本上使用我拥有的版本：dict.flatMap { $0.value }.flatMap { $0.map{ $0 } }，然后像上面那样初始化字典，它可以工作

【解决方案3】：

您可以将dict 转换为字典数组，然后使用reduce 合并键。

let values = dict.map {$0.value}.flatMap {$0}

let merged = values .flatMap{$0}.reduce([String:Int]()) { (previous, dict) in
    var newDict = previous
    if let lastValue = previous[dict.key] {
        let sum = lastValue + dict.value
        newDict[dict.key] = sum
    } else {
        newDict.updateValue(dict.value, forKey: dict.key)
    }
    return newDict
}

print(merged)

【讨论】：

【解决方案4】：

您可以使用字典uniquingKeysWith初始化器并使用加号方法来总结它们：

let result: Dictionary = .init(dict.flatMap(\.value), uniquingKeysWith: +)

【讨论】：