Wie verschiebt Arrays und Matrizen in Scala

Question

Angenommen, ich muss ein Array von ganzen Zahlen nach links und rechts verschieben.

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

Angenommen, ich auch eine Matrix von ganzen Zahlen verschieben müssen links und rechts

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

und nach oben und unten

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) 

Nun möchte Ich mag verallgemeinernshl und shr um sie für das Verschieben der Matrix nach oben und unten wiederzuverwenden. Wie würdest du das machen?

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)