修复replaceNth函数：嵌套列表中第n项替换问题

Let's break down what's going wrong with your current code and how to fix it. Your goal is to replace the nth element of a list with all elements from another list (instead of inserting the replacement list as a single element), right?

The Issue with Your Current Code #

Looking at your implementation:

replaceNth n newVal (x:xs) 
  | n == 0 = newVal:[xs] 
  | otherwise = [x]++replaceNth (n-1) newVal xs

The problem is in the n == 0 case: newVal:[xs] wraps your replacement list into a single element, so you end up with an extra level of nesting where you don't want one. For example, in your test case, this would insert [["0","1"],["8","8"]] as a single element instead of expanding it into two separate elements in the result.

Fixed Implementation #

Here's the corrected function, with added clarity and boundary handling:

-- Polymorphic version works for any list type
replaceNth :: Int -> [a] -> [a] -> [a]
replaceNth _ _ [] = []  -- Handle empty input list gracefully
replaceNth n newVal (x:xs)
  | n == 0    = newVal ++ xs  -- Replace current element with all elements from newVal
  | otherwise = x : replaceNth (n-1) newVal xs  -- Keep current element, recurse on the rest

Why This Works #

Let's walk through your test case to see it in action:
Input: replaceNth 2 [["0","1"],["8","8"]] [["2","3"],["1","2"],["3","4"],["9","12"]]

1. First call (n=2): Keep ["2","3"] and recurse with n=1 on the remaining list [["1","2"],["3","4"],["9","12"]]
2. Second call (n=1): Keep ["1","2"] and recurse with n=0 on the remaining list [["3","4"],["9","12"]]
3. Third call (n=0): Replace the current element (["3","4"]) by concatenating newVal with the rest of the list ([["9","12"]]), giving [["0","1"],["8","8"],["9","12"]]
4. Putting it all together: We get [["2","3"],["1","2"]] ++ [["0","1"],["8","8"],["9","12"]], which matches your expected output exactly.

Type-Specific Version (If Needed) #

If you only need it for nested string lists, you can use this explicit type signature instead:

replaceNth :: Int -> [[String]] -> [[String]] -> [[String]]
replaceNth _ _ [] = []
replaceNth n newVal (x:xs)
  | n == 0    = newVal ++ xs
  | otherwise = x : replaceNth (n-1) newVal xs

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