如何在Swift中获取2数组的公共元素列表？

Question

我有两个数组：

fruitsArray = ["apple", "mango", "blueberry", "orange"]
vegArray = ["tomato", "potato", "mango", "blueberry"]

我如何获得这两个数组中的常见项目列表

ouptput = ["mango", "blueberry"]

我无法使用，if contains(array, string)因为我想比较2个数组。

Answer 1

您还可以结合使用filter和contains：

let fruitsArray = ["apple", "mango", "blueberry", "orange"]
let vegArray = ["tomato", "potato", "mango", "blueberry"]

// only Swift 1
let output = fruitsArray.filter{ contains(vegArray, $0) }

// in Swift 2 and above
let output = fruitsArray.filter{ vegArray.contains($0) }
// or
let output = fruitsArray.filter(vegArray.contains)

我们考虑以下代码片段：

let array1: Array = ...
let array2: Array = ...

// `Array`
let commonElements = array1.filter(array2.contains)

// vs `Set`
let commonElements = Array(Set(array1).intersection(Set(array2)))
// or (performance wise equivalent)
let commonElements: Array = Set(array1).filter(Set(array2).contains)

我用Intshort和long Strings（10到100
Characters）（全部随机生成）做了一些（人工）基准测试。我总是用array1.count == array2.count

我得到以下结果：

如果您不只critical #(number of) elements转换为a，则更Set可取

data         |  critical #elements
-------------|--------------------
         Int |        ~50
short String |       ~100
 long String |       ~200

结果说明

使用该Array方法使用“蛮力”搜索，该搜索具有时间复杂度
O(N^2)，N = array1.count = array2.count而与该Set方法相反O(N)。然而从转换Array到Set和背部是这解释了增加大数据非常昂贵的`critical

elements`更大的数据类型。

结论

对于Array具有约100个元素的小s，该Array方法很好，但对于较大的s，则应使用该Set方法。

如果您想多次使用此“常见元素”运算，则建议 仅* 在可能的情况下使用Sets （元素的类型必须为）。 *Hashable

结束语

从Array到Set的转换比较昂贵，而从Set到的转换Array则非常便宜。

filter与with一起使用比在以下.filter(array1.contains)情况下性能更快.filter{ array1.contains($0) }：

• 最后一个创建一个新的闭包（ 仅一次 ），而第一个仅传递一个函数指针
• 对于最后一个封闭件的调用创建花费空间和时间的附加堆栈帧（ 多次 ：O(N)）