Как объединить значения в списки заданных длин в Haskell?

Question

Пытается написать функцию в Haskell, которая связывает список xs в списки размера n, например. результат bundle 3 [1..10] должен быть [[1,2,3],[4,5,6],[7,8,9],[10]].

Я знаю, что тип должен быть

bundle :: Int -> [a] -> [[a]] 

и должен удовлетворять что-то вроде

concat $ bundle n xs == xs 
length xss > 1 ==> all (\xs -> n == length xs) (init xss) 

, но когда я приезжаю на самом деле попытаться осуществить его я застрял. Я предполагаю, что мне нужно использовать foldl, но не могу придумать, какую функцию применять.

Answer 1

Ну, вы можете использовать splitAt, например.

bundle :: Int -> [a] -> [[a]] 
bundle _ [] = [] 
bundle n xs = as : bundle n bs 
    where (as, bs) = splitAt n xs 

однако, вы должны убедиться, что n является положительным, в противном случае вы будете в конечном итоге с бесконечным списком пустых списков. Эта функция сохраняет ваши свойства: splitAt n возвращает пару, где первая часть кортежа имеет точно n элементов. Поскольку все, кроме последней записи списка, создаются таким образом, ваше второе свойство сохраняется. Первое свойство выполняется по понятным причинам.

Однако эти «пучки» обычно называются «кусками». Пакет split обеспечивает функцию фитинг chunksOf, который делает точно то же самое:

import Data.List.Split (chunksOf) 

bundle :: Int -> [a] -> [[a]] 
bundle = chunksOf