Haskell 如何基于相邻元素total字段差值计算new字段值

核心问题解答 #

能不能单独用map实现？ #

不能。map 只能对列表中的单个元素做独立转换，无法访问相邻的前序元素的值，因此满足不了需要对比前一个total计算new的需求。

可行实现方案 #

有两种常用实现方式，不需要写命令式的for循环，两种方案运行后都可以得到你给出的「处理后」表格的效果：

方案1：标准库高阶函数组合（推荐）

用zipWith配对当前元素和前序元素的total计算差值，代码最简洁：

data Item = Item{
    date :: Day,
    total :: Int,
    new :: Int
} deriving (Show)

calcNew :: [Item] -> [Item]
calcNew [] = []
calcNew allItems@(first:rest) = 
    let processedFirst = first { new = 0 }
        totalList = map total allItems
        -- 逐个计算后续元素的new值
        newValues = zipWith (-) (tail totalList) totalList
        -- 给剩余元素赋值new
        processedRest = zipWith (\item val -> item { new = val }) rest newValues
    in processedFirst : processedRest

方案2：手动递归实现

如果想要更直观的控制遍历逻辑，可以用递归传递前一个元素的total值：

calcNewRec :: [Item] -> [Item]
calcNewRec [] = []
calcNewRec (first:rest) = first { new = 0 } : traverseRest rest (total first)
    where
        traverseRest [] _ = []
        traverseRest (current:remaining) prevTotal =
            let currentNew = total current - prevTotal
            in current { new = currentNew } : traverseRest remaining (total current)

原有代码问题说明 #

你之前定义的calcNew签名多了不需要的Int参数，new值可以完全通过输入列表的元素计算得到，不需要外部传入；updateOneItem的功能可以直接用Haskell原生的记录更新语法item { new = 待赋值 }代替，无需单独封装函数。

内容的提问来源于stack exchange，提问作者user142650