使用cons(::)而不是concat(@)构建列表时F#折叠

时间:2014-02-26 09:06:39

标签: list f# concat fold

我有以下功能可以满足我的需求。但是使用的是concat(@)运算符,它是O(n)而不是O(1)的(::)运算符

let myFunc s m cs =  
    let n = s * m
    let c = [n - s] // single element list
    (n,  cs @ c) // concat the new value to the accumulated list

let chgLstAndLast = 
    [0.99; 0.98; 1.02] 
    |> List.fold (fun (s, cs) m -> myFunc s m cs) (1., []) 

chgLstAndLast返回最后生成的结果值和列表:

val chgLstAndLast : float * float list = (0.989604, [-0.01; -0.0198; 0.019404])

我想以三种方式改进上述内容。

  1. 使用con(::)而不是concat(@)
  2. 将列表累积从myFunc移动到List.fold操作
  3. 确保结果列表顺序与上面相同(即最后一个结果位于列表末尾而不是头部)
  4. 例如,我想像这样写一个myFunc

    let myFunc s m cs =  
        let n = s * m
        let c = n - s // single element, but not as list
        (n, c) // No concat here
    

    但是当我这样做时,我没有看到如何在折叠功能中使用(::)缺点。

1 个答案:

答案 0 :(得分:6)

如果我正确理解您的代码,您要做的是fold同时保留所有中间结果。这几乎是List.scan的作用;它也会返回初始状态。

let chgLstAndLast data =
    let inner s m =
        let n = s * m
        n, n - s
    let processedData = data |> List.scan (fun (s, _) n -> inner s n) (1.0, 1.0)
    let lastResult = processedData |> List.reduce (fun _ n -> n)
    let seq = processedData |> List.tail |> List.map snd
    lastResult, seq

要详细解释这段代码:首先我声明一个内部函数,使代码对外部世界更清晰(假设其他代码不需要myFunc),然后我使用{{ 1}}从scan获取所有中间结果,这是一种内置的折叠+累加器技巧。
最后一个值是通过fold技巧获得的,因为没有内置的"列表的最后一个"函数,中间结果是处理数据的第二部分,除了第一个元素是初始状态。

相关问题