Swift中定义通用泛型unfoldr函数报类型不匹配错误如何解决

你遇到类型不匹配问题，同时原实现存在逻辑缺陷，核心原因是误用了Swift标准库的序列构造方法：

• 你当前调用的sequence(first:next:)是专门为「产出元素和中间状态完全同类型」的场景设计的，它的签名要求first参数、next闭包的输入输出类型完全一致，没有独立的状态存储位：

  func sequence<T>(first: T, next: @escaping (T) -> T?) -> UnfoldFirstSequence<T>
  

• 你要实现的unfoldr逻辑，核心是产出元素类型A、中间状态类型B相互独立的：一方面你传入的初始参数b是状态类型B，却被传给了要求元素类型A的first参数，直接触发了类型不匹配的编译错误；另一方面就算A和B碰巧类型相同，原实现也丢失了每次f返回的下一个状态值b，导致next闭包下一次拿到的是上一个产出的元素a，而非正确的状态值，逻辑是完全错误的。

Swift标准库提供了专门支持「状态与产出元素类型分离」的序列构造方法sequence(state:next:)，完全适配unfoldr的设计逻辑，对应实现如下：

func unfoldr<A, B>(_ f: @escaping (B) -> (A, B)?) -> (B) -> UnfoldSequence<B, A> {
    return { initialState in
        sequence(state: initialState, next: { currentState in
            // 用当前状态调用传入的转换函数
            guard let (element, nextState) = f(currentState) else {
                return nil // 返回nil终止序列
            }
            // 把状态更新为下一次的值
            currentState = nextState
            // 返回本次要产出的元素
            return element
        })
    }
}

可以用斐波那契数列生成的场景验证可用性：

// 定义斐波那契生成逻辑：状态为相邻两个数的元组
let fibGenerator = unfoldr { (state: (Int, Int)) -> (Int, (Int, Int))? in
    let (current, next) = state
    // 产出当前数，下一个状态为(next, current+next)
    return (current, (next, current + next))
}

// 从初始状态(0,1)开始，取前10个元素
Array(fibGenerator((0, 1)).prefix(10))
// 输出: [0, 1, 1, 2, 3, 5, 8, 13, 21, 34]

实现返回的UnfoldSequence<B, A>完全遵循Swift的Sequence协议，可以正常使用所有序列内置方法（map、filter、reduce等），不会有类型兼容问题。

内容的提问来源于stack exchange，提问作者F. Zer