从子数组中获取单个项目的快速功能方法

Question

我有以下功能(示例)，我想以功能方式编写:

func passengerForId(_ id: Int) -> Passenger?
{
    for car in cars
    {
        for passenger in car.passengers
        {
            if passenger.id == id
            {
                return passenger
            }
        }
    }
    return nil
}

什么是最好的方法？

我试过以下方法:

func passengerForId(_ id: Int) -> Passenger?
{
    return cars.first(where: { $0.passengers.contains(where: { $0.id == id } ) })
}

但是，这显然会返回包含所请求乘客的汽车，而不是乘客本身。

另一种选择是这样的:

func passengerForId(_ id: Int) -> Passenger?
{
    let passengers = cars.flatMap { car in car.passengers }
    return passengers.first(where: { $0.id == id })
}

或

func passengerForId(_ id: Int) -> Passenger?
{
    return cars.flatMap { car in car.passsengers.filter { $0.id == id }}.first
}

但是，这似乎比原来的效率低，因为它会遍历所有区域。

有更好的方法吗？

Answer 1

您可以创建一个(懒惰的)所有乘客的扁平集合，然后选择 第一个匹配的:

func passengerForId(_ id: Int) -> Passenger? {
    return cars.lazy.flatMap({ $0.passengers }).first(where: { $0.id == id } )
}