OCaml: перестановка каждого значения в двух наборах? (как перевести это с Java) - PullRequest
4 голосов
/ 02 октября 2009

У меня есть два набора, возвращаемых Set.Make (t). Я хотел бы генерировать каждую возможную комбинацию значений в двух. Как я могу это сделать?

Это работает для генерации некоторых пар, но не всех:

List.combine (IntSet.elements odp) (IntSet.elements ftw)

Это будет сделано в Java:

for (int i : ktr) {
     for (int m : mnx) {
       System.out.println(i + ", " + m);
     }
}

Ответы [ 3 ]

6 голосов
/ 02 октября 2009

Объединение решения @David Crawshaw (которое является хвост-рекурсивным) с решением @ newacct (которое является полностью универсальным):

let cartesian_product xs ys =
  List.fold_left (fun acc x -> 
    List.fold_left (fun acc y -> 
      (x,y) :: acc) 
      acc ys) 
    [] xs

let product = 
  cartesian_product (IntSet.elements odb) (IntSet.elements ftw)

Это обратит естественный порядок. Его можно восстановить, применив List.rev к результату (List.rev также является хвост-рекурсивным).

5 голосов
/ 02 октября 2009

Если xs и ys - два списка, то их декартово произведение (возвращающее список пар) может быть вычислено следующим образом:

List.concat (List.map (fun x -> List.map (fun y -> (x, y))
                                         ys)
                      xs)

В этом случае ваши xs и ys равны IntSet.elements odp и IntSet.elements ftw

3 голосов
/ 02 октября 2009

Вы ищете декартово произведение двух наборов.

Этот вопрос был задан в теме в списке рассылки OCaml. Этот ответ предлагает Брайан Херт : Для

module TypeSet = Set.Make(Type);;

Создать, чтобы представить продукт:

module TypeType = struct
    type t = Type.t * Type.t;;
    let compare (x1, x2) (y1, y2) =
        let r = Type.compare x1 y1 in
        if r == 0 then
            Type.compare x2 y2
        else
            r
    ;;
end;;

module TypeTypeSet = Set.Make(TypeType);;

Затем сгенерируйте продукт с помощью:

let cartesian_product s1 s2 =
    let f x y t = TypeTypeSet.add (x, y) t in
    let g x t = TypeSet.fold (f x) s2 t in
    TypeSet.fold g s1 TypeTypeSet.empty
;;
...