带有泛型参数或 T 或 T 数组的 Swift 重载函数中的意外行为

Question

我正在尝试通过泛型实现平面地图。 （是的，flatmap 已经是该语言的一部分。）我的尝试使用通用的重载函数来实现这一点。 （可能不是最有效的。我知道。我已经有了一个有效的命令式版本。）我发现出乎意料的是 flatten< T, U >( _ elem: T ) -> [ U ] 正在消耗调用 flatten 的数组，即使存在重载 func flatten< T, U >( _ ary: [ T ] ) -> [ U ] 也是如此。

我可以知道出了什么问题，但我似乎看不出原因，甚至看不出去哪里了解原因并可能修复它。

我什至尝试了一些演员表，只是为了看看，结果产生了错误，这对我来说很有意义。为什么要将T 转换为T 或[ T ] 转换为[ T ]？。还将ary 更改为elem，以查看这是否导致重载语义问题。产生相同的输出。

import Foundation

func flatten< T, U >( _ elem:    T   ) -> [ U ] {
    if let item = elem as? U {
        return [ item ]
    } else {
        return []
    }
}

func flatten< T, U >( _ ary:   [ T ] ) -> [ U ] {
    if ary.count == 0 {
        return []
    }

    return flatten( ary.first ) + flatten( Array( ary.dropFirst() ) )
}

let x: [ Int ] = flatten( [ 1, 2, [ 3, 4 ], [ [ 5, 6 ], nil, 7 ], 8, 9 ] )
print( x )  // Prints [1, 2, 8, 9 ] instead of [1, 2, 3, 4, 5, 6, 7, 8, 9]

Answer 1

编译器对使用从 ary.first 出来的封装可选选项感到困惑，这肯定会转到第一个函数签名。但即使修复了这个问题，仍然会在处理更简单的情况（例如 [[1,2][3,4]]）时留下一些歧义。

解决这个问题的另一种方法是在内部处理所有类型，而不是将其留给编译器：

func flatten<U>( _ ary:[Any] ) -> [ U ]
{
    func makeFlat(_ flat:[U], _ elem:Any) -> [U]
    {
       if let item   = elem as? U     { return flat + [item] }
       if let ary    = elem as? [U]   { return flat + ary }
       if let subAry = elem as? [Any] { return flat + flatten(subAry) }
       return flat
    }
    return ary.reduce([U](),makeFlat)
}