你如何用Scala中的案例进行记忆?

时间:2015-01-17 20:51:46

标签: scala dynamic-programming memoization

使用案例和函数式编程将使用memoization的代码转换为适当的Scala的最佳方法是什么?

def uniquePathsMemoization(n:Int, m:Int, row:Int, col:Int, seen:Array[Array[Int]]):Int = {
  if (row == m && col == n) 1
  if (row > m || col > n) 0

  if (seen(row+1)(col) == -1) seen(row+1)(col) = uniquePathsMemoization(n, m, row + 1, col, seen)
  if (seen(row)(col + 1) == -1 ) seen(row)(col) = uniquePathsMemoization(n,m, row, col + 1, seen)

seen(row+1)(col) + seen(row)(col + 1)
}

1 个答案:

答案 0 :(得分:1)

这是使用matchcase

的代码的修改版本
def uniquePathsMemoization(n:Int, m:Int, row:Int, col:Int, seen:Array[Array[Int]]):Int = (row,col) match{

  case (row,col) if row == m && col == n =>
    1
  case (row,col) if row > m || col > n =>
    0
  case (row,col) =>
    if (seen(row+1)(col) == -1) seen(row+1)(col) = uniquePathsMemoization(n, m, row + 1, col, seen)
    if (seen(row)(col + 1) == -1 ) seen(row)(col) = uniquePathsMemoization(n,m, row, col + 1, seen)

    seen(row+1)(col) + seen(row)(col + 1)
}

由于存储在seen数组中的状态,将此代码转换为纯函数版本并不容易。但是使用函数装饰器可以为应用程序的其余部分隐藏此状态:

def uniquePathsMemoizationGenerator( maxRows: Int, maxCols:Int ) : (Int,Int,Int,Int) => Int = {


  def uniquePathsMemoization(n:Int, m:Int, row:Int, col:Int, seen:Array[Array[Int]]):Int = (row,col) match{

    case (row,col) if row == m && col == n =>
      1
    case (row,col) if row > m || col > n =>
      0
    case (row,col) =>
      if (seen(row+1)(col) == -1) seen(row+1)(col) = uniquePathsMemoization(n, m, row + 1, col, seen)
      if (seen(row)(col + 1) == -1 ) seen(row)(col) = uniquePathsMemoization(n,m, row, col + 1, seen)

      seen(row+1)(col) + seen(row)(col + 1)
  }

  val seen = Array.fill(maxRows,maxCols)(-1)
  uniquePathsMemoization(_,_,_,_,seen)
}

val maxRows = ???
val maxCols = ???
val uniquePaths = uniquePathsMemoizationGenerator( maxRows, maxCols )

// Use uniquePaths from this point, instead of uniquePathsMemoization