1

我是 OCaml 的初学者，我想从文件中读取行，然后检查每行中的所有字符。作为一个虚拟示例，假设我们要计算文件中字符“A”的出现次数。

我尝试了以下

open Core.Std

let count_a acc string = 
    let rec count_help res stream =
        match Stream.peek stream with
        | None -> res
        | Some char -> Stream.junk stream; if char = 'A' then count_help (res+1) stream else count_help res stream
    in acc + count_help 0 (Stream.of_string string)

let count_a = In_channel.fold_lines stdin ~init:0 ~f:count_a

let () = print_string ((string_of_int count_a)^"\n"

我用它编译

 ocamlfind ocamlc -linkpkg -thread -package core -o solution solution.ml

运行它

$./solution < huge_file.txt

在一个有一百万行的文件上，这给了我以下时间

real    0m16.337s
user    0m16.302s
sys 0m0.027s

这是我的python实现的4倍。我很确定应该可以让它更快，但我应该怎么做呢？

4

1 回答 1

3

要计算字符串中 A 字符的数量，您可以使用String.count函数。事实上，最简单的解决方案将是：

open Core.Std

let () =
  In_channel.input_all stdin |>
  String.count ~f:(fun c -> c = 'A') |>
  printf "we have %d A's\n"

更新

使用 [fold_lines] 的稍微复杂一点的解决方案（内存消耗更少）如下所示：

let () =
  In_channel.fold_lines stdin ~init:0 ~f:(fun n s ->
    n + String.count ~f:(fun c -> c = 'A') s) |>
    printf "we have %d A's\n"

事实上，它比前一个慢。在我用了 8 年的笔记本电脑上，计算 20 兆字节文本文件中的“A”需要 7.3 秒。前一个解决方案需要 3 秒。

另外，我希望你会发现这篇文章很有趣。

于 2014-10-01T15:35:25.890 回答