OCaml中的螺旋印刷

时间:2015-04-05 02:36:38

标签: clojure ocaml

有着名的练习以螺旋方式打印矩阵(表示为矢量矢量),我想在OCaml中进行。

鉴于以下矩阵,应打印1 2 3 4 5 6 7 8 9

[ [1 2 3] 
  [8 9 4] 
  [7 6 5] ] 

优雅的实现将第一个子向量添加到累加器,并使用旋转的剩余数字递归调用该函数。

旋转的剩余部分在第一次迭代后看起来如下

[ [4 5] 
  [9 6] 
  [8 7] ]

在Clojure中,我执行以下操作来旋转矢量:

(reverse (apply map vector [[8 9 4] [7 6 5]]))
([4 5] [9 6] [8 7])

我想知道OCaml中是否有类似的东西。我找到了List.map但不确定其余的。我应该只实现从每个子向量中读取最后一个元素的东西吗?

1 个答案:

答案 0 :(得分:4)

以下是旋转功能的可能实现:

let rotate ll =
    let cons x y = x :: y in
    let init = List.map (fun _ -> []) (List.hd ll) in
    let rres = List.fold_right (List.map2 cons) ll init in
    List.rev rres

以下是一个示例会话:

$ ocaml
        OCaml version 4.01.0

# #use "rot.ml";;
val rotate : 'a list list -> 'a list list = <fun>
# rotate [[8; 9; 4]; [7; 6; 5]];;
- : int list list = [[4; 5]; [9; 6]; [8; 7]]