使用 Data.Map.adjust 折叠

编程问答 2022-05-25

问题描述

我正在尝试获得正确的函数，该函数将通过迭代键列表来更新一个或多个值。

示例：

s1 = Data.Map.fromList [(1,"value1"),(2,"value2"),(3,"value3")]

通过将列表 [1,2] 和更新值 "plus" 传递给函数，列表应如下所示：

[(1,"plus"),"value3")]

这是我到目前为止所尝试的，我很难实现迭代要更新的键列表的函数

changeValue map newValue xs = foldr (\k map -> Map.adjust (newValue)(xs)(sm)) map m

我应该从那里寻找什么？我认为在这种情况下我不能使用 map。

解决方法

好的，你走对了。

折叠确实是您所需要的。您需要折叠密钥列表。起始值是您的原始地图，它在折叠期间正在转换。

您也正确识别了功能调整。

adjust 需要一个函数作为第一个参数。这个函数是\_ -> newValue，或者就是const newValue。我不会使用名称 map，因为它可能会引起混淆。

经过这些更正后，您的程序可能看起来像这样：

import qualified Data.Map.Strict as M

s1 = M.fromList [(1,"value1"),(2,"value2"),(3,"value3")]

changeValue m newValue ks = foldr (\k m -> M.adjust (const newValue) k m) m ks

main :: IO ()
main = do
    print $ changeValue s1 "plus" [1,2]
dictionaryfoldfoldhaskellhaskelliterationiteration