将一个函数应用于haskell列表中的第n项n次

applyNTimes :: (a -> a) -> Int -> [a] -> [a]
applyNTimes _ _ [] = []
applyNTimes f 0 xs = xs
applyNTimes f n (x:xs) = x : applyNTimes f (n-1) (f x : xs)

-- 将元素加倍3次
doubleNTimes :: Num a => [a] -> [a]
doubleNTimes xs = applyNTimes (*2) 3 xs

-- 将元素平方2次
squareNTimes :: Num a => [a] -> [a]
squareNTimes xs = applyNTimes (^2) 2 xs

-- 示例输出
main :: IO ()
main = do
  let myList = [1, 2, 3, 4, 5]
  putStrLn "Original list:"
  print myList
  putStrLn "After doubling 3 times:"
  print (doubleNTimes myList)
  putStrLn "After squaring 2 times:"
  print (squareNTimes myList)

Original list:
[1,2,3,4,5]
After doubling 3 times:
[1,2,4,8,16,32]
After squaring 2 times:
[1,4,16,256,65536]