Scalaで配列と行列をシフトする方法

Question

整数の配列を左右にシフトする必要があるとします。

val shl : Array[Int] => Array[Int] = a => (a drop 1) :+ 0 
val shr : Array[Int] => Array[Int] = a => 0 +: (a dropRight 1) 

も、私は今、私はshlとshrを一般化したい

val shlm: Array[Array[Int]] => Array[Array[Int]] = _ map shl 
val shrm: Array[Array[Int]] => Array[Array[Int]] = _ map shr 

上下

val zeros: Int => Array[Int] = n => Array.fill(n)(0) 
val shum: Array[Array[Int]] => Array[Array[Int]] = m => (m drop 1) :+ zeros(m.length) 
val shdm: Array[Array[Int]] => Array[Array[Int]] = m => zeros(m.length) +: (m dropRight 1) 

を左右する整数の行列をシフトする必要があるとし行列を上下にシフトするためにそれらを再利用する。あなたはどうしますか？

Answer 1

import scala.reflect.ClassTag 

type ArrayOp[T] = Array[T] => Array[T] 

def shl[T:ClassTag](zero: T)(a: Array[T]): Array[T] = (a drop 1) :+ zero 
def shr[T:ClassTag](zero: T)(a: Array[T]): Array[T] = zero +: (a dropRight 1) 

val shlm: ArrayOp[Array[Int]] = _ map shl(0) 
val shrm: ArrayOp[Array[Int]] = _ map shr(0) 

val zeros: Int => Array[Int] = n => Array.fill(n)(0) 

val shum: ArrayOp[Array[Int]] = m => shl(zeros(m.length))(m) 
val shdm: ArrayOp[Array[Int]] = m => shr(zeros(m.length))(m)