Swift 2计数重复数组

时间:2015-12-01 21:18:40

标签: arrays swift count duplicates

你能帮我简化下面的流程吗? 范围是计算给定数组中每个元素重复的时间。

var test = [1,2,3,4,5,6,7,8,5,9,3,3,9,9,9]

var testCount = [Int:Int]()

for curr in test {
    if let x = testCount[curr] {
        testCount[curr] = x  + 1
        continue;
    }
    testCount[curr] = 1
}

print(testCount)

我正在努力,以便有类似的东西:

test.map_duplicate() or map_duplicate(test)

3 个答案:

答案 0 :(得分:4)

extension Array where Element: Hashable {

    func mergeDuplicates() -> [Element: Int] {
        var result = [Element: Int]()

        self.forEach({ result[$0] = result[$0] ?? 0 + 1 })

        return result
    }

}

var test = [1,2,3,4,5,6,7,8,5,9,3,3,9,9,9]

let testCount = test.mergeDuplicates()

print(testCount)

上面的代码为任何Array类型提供了扩展,其中包含Hashable个元素(其他任何内容都不符合字典的键)。方法mergeDuplicates生成您想要的字典。在其实现中,forEach()对数组中的每个元素执行闭包(这通常是对老式for循环更直接和透明的替代)。然后在每次迭代时,我们检查字典中是否已存在具有运行计数的值(并使用nil-coalescing运算符??来处理之前尚未满足特定值的情况,因此它是零)。然后递增,并存储(返回)字典。

答案 1 :(得分:3)

你可以扩展SequenceType以获得一个函数freq,它完全符合你的需要。

extension SequenceType where Self.Generator.Element: Hashable {
    func freq() -> [Self.Generator.Element: Int] {
        return reduce([:]) { (var accu: [Self.Generator.Element: Int], element) in
            accu[element] = accu[element]?.successor() ?? 1
            return accu
        }
    }
}

答案 2 :(得分:0)

var countForNumber = 0

for number in test {
    test.forEach({ if number == $0 { countForNumber++ } })
    testCount[number] = countForNumber
    countForNumber = 0
}