映射列表,最后一个列表元素除外

Question

除了最后一个列表元素外,如何最好地映射列表的所有元素？

假设我们有一个列表let l = [1,2,3,4]并希望得到[2,3,4,4].

我确实有一个解决方案,但它感觉不像是"功能"的方式(在ghci中):

let l = [1,2,3,4]
let len = toIntegral $ length l -- to avoid a type mismatch Integer <-> Int
let l1 = zip l [1..]
let l2 = map (\(x,y) -> if y < len then (x + 1,y) else (x,y)) l1
let l3 = map (fst) l2

不太好......我希望有更好的方法!由于我是函数式编程的新手,我不知道从哪里开始寻找它.

Answer 1

只需重写,map但只有一个元素时才会出现特殊情况:

mapBut1 :: (a -> a) -> [a] -> [a]
mapBut1 f [] = []
mapBut1 f [x] = [x]
mapBut1 f (x:xs) = f x : mapBut1 f xs

现在,即使对于无限列表也是如此,它比计算长度快得多,并使其更具可读性.请注意,这确实将您的函数限制为类型a -> a而不是a -> b.

或者,你可以做到

mapBut1 f (x:y:xs) = f x : mapBut1 f (y:xs)
mapBut1 f other = other

它们是等价的定义,但后者使用的模式匹配减少1.不过,我更喜欢前者,因为处理案件的情况更为明显.

Answer 2

像往常一样,这是假装 - paramorphism的工作:

import Data.List (tails)

mapButLast :: (a -> a) -> [a] -> [a]
mapButLast f = foldr g [] . tails where
  g (x:_:_) r = f x : r
  g xs      _ = xs

或者正确地说para我们只是写

mapButLast f = para g [] where
  g x [] r = [x]
  g x _  r = f x : r

哪里

para f z (x:xs) = f x xs (para f z xs)
para f z []     = z