清除通过唯一字典值过滤字典数组的方法

时间:2017-03-01 21:05:10

标签: arrays swift dictionary filter

说我有一个对象:

struct Foo {
    let id: Int
    let bar: Int
}

现在我在数组中有5个这样的对象:

let foo1 = Foo(id: 1, bar: 1)
let foo2 = Foo(id: 2, bar: 1)
let foo3 = Foo(id: 3, bar: 2)
let foo4 = Foo(id: 4, bar: 3)
let foo5 = Foo(id: 5, bar: 3)

let fooArray = [foo1, foo2, foo3, foo4, foo5] 

过滤具有唯一foo值的bar个对象的简洁方法是什么?

// Desired output
let filteredArray = [foo1, foo3, foo4]

假设有几百到几千个对象可以迭代。

2 个答案:

答案 0 :(得分:5)

一种可能的方法是使用Set来跟踪哪个 已经看到bar值:

var seenBarValues = Set<Int>()
let filteredArray = fooArray.filter { foo in
    if seenBarValues.contains(foo.bar) {
        // We already had a `Foo` with this `bar` value: skip.
        return false 
    } else {
        // First `Foo` with this `bar` value: remember and include.
        seenBarValues.insert(foo.bar)
        return true
    }
}

正如@Hamish正确指出的那样,这可以缩短为

var seenBarValues = Set<Int>()
let filteredArray = fooArray.filter { 
    seenBarValues.insert($0.bar).inserted
}

使用

这一事实
public mutating func insert(_ newMember: Element) -> (inserted: Bool, memberAfterInsert: Element)

返回一个元组,其第一个成员指示元素是否相等 新插入的那个已经出现在集合中。

答案 1 :(得分:0)

我不是公认答案的粉丝。将集合放在使用它的闭包之外似乎不合适。我宁愿保留一切。有一个CS术语适用,但我不记得它是什么...

我宁愿看到这个:

let uniquedBars = fooArray
    .reduce(into: (result: [Foo](), set: Set<Int>())) { partial, next in
        if partial.set.insert(next.bar).inserted {
            partial.result.append(next)
        }
    }
    .result