我的数组中有大约 2000 个元素,当它被过滤时,我想在过滤后的数组中有 5 个元素后立即结束过滤。
目前是:
providerArray.filter({($0.lowercased().range(of:((row.value as? String)?.lowercased())!) != nil)})
它可以返回多达 2000 个结果,这是浪费处理和时间。
为了更清楚,我需要一个解决方案,它等效于限制过滤器结果,就像我可以使用 coreData 获取[request setFetchLimit:5];
就执行时间而言,最快的解决方案似乎是显式循环,添加匹配元素,直到达到限制:
extension Sequence {
public func filter(where isIncluded: (Iterator.Element) -> Bool, limit: Int) -> [Iterator.Element] {
var result : [Iterator.Element] = []
result.reserveCapacity(limit)
var count = 0
var it = makeIterator()
// While limit not reached and there are more elements ...
while count < limit, let element = it.next() {
if isIncluded(element) {
result.append(element)
count += 1
}
}
return result
}
}
用法示例:
let numbers = Array(0 ..< 2000)
let result = numbers.filter(where: { $0 % 3 == 0 }, limit: 5)
print(result) // [0, 3, 6, 9, 12]
您也可以使用.lazy
稍微提高性能:
let numbers: [Int] = Array(0 ..< 2000)
let result: AnySequence = numbers
.lazy
.filter {
print("Calling filter for: ($0)")
return ($0 % 3) == 0
}
.prefix(5)
print(Array(result))
这将仅对前 15 个值调用 filter
函数(直到找到 5 个通过筛选器的值)。
现在,您可以专注于提高filter
本身的性能。 例如,通过缓存值。您不必这样做,但如果某些值不断重复,它可以大大提高性能。
let numbers: [Int] = Array(0 ..< 2000)
var filterCache: [Int: Bool] = [:]
let result: AnySequence = numbers
.lazy
.filter {
if let cachedResult = filterCache[$0] {
return cachedResult
}
print("Calling filter for: ($0)")
let result = (($0 % 3) == 0)
filterCache[$0] = result
return result
}
.prefix(5)
print(Array(result))
可以将此方法直接应用于函数。
另请注意,要提高性能,您应该:
将
((row.value as? String)?.lowercased())!
保存到局部变量中,因为它被多次执行使用选项简化表达式:
let result: AnySequence = providerArray
.lazy
.filter {
$0.range(of: row.value as! String, options: [.caseInsensitive]) != nil
}
.prefix(5)