自定义OCaml List.flatten后出现类型匹配错误,请求排查
问题分析与解决
首先明确:错误根源不是custom_flatten函数本身,而是generate_transfers函数中对mapi_Two的调用参数顺序错误,替换标准库flatten后类型推断路径变化,触发了这个隐藏问题。
错误原因
看mapi_Two的定义:
let rec mapi_Two f index list = match list with | [] -> [] | x :: xs -> let result = f index x in (* 先传整数索引index,再传列表元素x *) result :: mapi_Two f (index + 1) xs
它要求传入的函数f第一个参数是整数索引,第二个参数是列表中的元素。但你在generate_transfers里的调用是:
|> mapi_Two (fun (source, target) i -> Transfer (source, target))
这里把匿名函数的第一个参数写成了元组(source, target)(对应列表元素),第二个参数写成了索引i,完全颠倒了mapi_Two的参数传递顺序,导致编译器报错:期望接收int类型的索引,却得到了元组类型的值。
修复方案
调整匿名函数的参数顺序,让第一个参数接收索引,第二个参数接收元组元素:
|> mapi_Two (fun i (source, target) -> Transfer (source, target))
mapi_Two会先传入索引值,再传入列表中的每个元组元素,这样参数类型就完全匹配了。
验证custom_flatten的正确性
你的custom_flatten实现是正确的,和标准库List.flatten功能一致:
- 辅助函数
aux通过递归把嵌套列表的元素依次加入累加器(逆序) - 最后用
reversed反转累加器得到正确顺序的扁平列表 - 测试示例:
custom_flatten [[Fill 0; Drain 0]; [Transfer (0,1)]]会返回[Fill 0; Drain 0; Transfer (0,1)],符合预期
内容的提问来源于stack exchange,提问作者lucas lootan
相关产品推荐
相关产品推荐

