swift - 单行 : create Dict key or update count in swift

Question

似乎应该有一个简单的方法来做到这一点:

for each token
   look it up in a dictionary
   if it's there already, increment the value by 1
   else create it and set value to 1

我可以用

for token in tokens {
    if let count = myDict[token] {
        myDict[token] = count + 1
    } else {
        myDict[token] = 1
}

但似乎必须有一种更优雅的单行方式来做到这一点？

Answer 1

您可以使用三元运算符:

for token in tokens {
    myDict[token] = myDict[token] ? myDict[token]! + 1 : 1
}

更好的是，使用零合并:

for token in tokens {
    myDict[token] = (myDict[token] ?? 0) + 1
}

然后将整个内容放在一行中:

tokens.forEach { myDict[$0] = (myDict[$0] ?? 0) + 1  }

使用 Swift 4(感谢 Hamish)，它可以更短一些:

tokens.forEach { myDict[$0, default: 0] += 1 }