创建直方图OCaml

时间:2016-11-05 19:49:04

标签: ocaml

我的任务是创建一个直方图,输出它在列表中的元素的次数。

Input:[2;2;2;3;4;4;1]
Output[(2, 3); (2, 2); (2, 1); (3, 1); (4, 2); (4, 1); (1, 1)]  
Expected output : [(2, 3); (3, 1); (4, 2); (1, 1)]

My code:

let rec count a ls = match ls with
  |[]              -> 0
  |x::xs  when x=a -> 1 + count a xs
  |_::xs           -> count a xs

let rec count a = function
  |[]              -> 0
  |x::xs  when x=a -> 1 + count a xs
  |_::xs           -> count a xs

let rec histo l = match l with
|[] -> []
|x :: xs ->  [(x, count x l)] @ histo xs ;;

我做错了什么?

1 个答案:

答案 0 :(得分:2)

问题是xs包含等于x的潜在元素。这就是你在你的输出中看到的:(2,3)意味着列表中有3次2;然后xs等于[2; 2; 3; 4; 4; 1] ......等等。

另外(不影响结论):你有2个计数定义,但它们是相同的。

要实现直方图,请使用Hashtbl:

let h = Hashtbl.create 1000;;    
List.iter (fun x -> let c = try Hashtbl.find h x with Not_found -> 0 in Hashtbl.replace h x (c+1)) your_list;;
Hashtbl.fold (fun x y acc ->  (x,y)::acc) h [];;