F#でHaskellのgroupByを作る

はじめに

以前書いた記事の続きです。
次は、HaskellのData.Listライブラリには関数"groupBy"を作ります。この関数は、リスト内の隣接する値同士を条件をもとにリストのグループを作成するものです。

コード

// 条件をもとにリスト内にリストのグループを作成する
let groupBy (predicate) (list: seq<'a>) =
    let rec auxAccumulate acc = function 
        | [] -> (acc, [])
        | x :: xs when predicate (List.last acc) x -> auxAccumulate (acc @ [x]) xs
        | xs -> (acc, xs)
    let rec auxProcess = function 
        | [] -> []
        | x::xs ->
            let (head,tail) = auxAccumulate [x] xs
            head :: auxProcess tail
    auxProcess (Seq.toList list)

実行結果

groupBy (<=) "abcabcaaa" |> printfn "%A"
//実行結果: [['a'; 'b'; 'c']; ['a'; 'b'; 'c']; ['a'; 'a'; 'a']]
groupBy (<=) [1;2;3;1;2;0;4;5;2] |> printfn "%A"
//実行結果: [[1; 2; 3]; [1; 2]; [0; 4; 5]; [2]]
groupBy (<) [1;2;2;3;3;3;1] |> printfn "%A"
//実行結果: [[1; 2]; [2; 3]; [3]; [3]; [1]]
groupBy (>) [|1;2;3;1;2;3|] |> printfn "%A"
//実行結果: [[1]; [2]; [3; 1]; [2]; [3]]