2009-09-12 19 views
7

Soy un principiante de Scala, que recién comienza a aprender el idioma.Formas de mejorar este código

Solucioné Problem 8 de la página Proyecto Euler.

El código es el siguiente (I eliminado todo el código que ver con la lectura de un archivo de entrada):

 
def max(n1: Int, n2: Int): Int = Math.max(n1, n2) 

def max_product(digits: List[Int], num: Int): Int = { 
    def max_core(lst: List[Int], curr_max: Int): Int = lst match { 
     case a if lst.length >= num => 
      max_core(a.tail, max(lst.slice(0, num).reduceLeft(_*_), curr_max)) 
     case _ => curr_max 
    } 

    max_core(digits, 0) 
} 

println(max_product(1::2::3::4::2::3::Nil, 2)) 

Funciona bien, el resultado es correcto. Sin embargo, no estoy completamente satisfecho con esta solución. No me gusta la subfunción max_core y tengo la sensación de que se puede mejorar. Mi entendimiento de FP es que deberías iterar sobre una lista, cortar no parece ser el camino aquí.

La pregunta es: ¿cómo?

Respuesta

7

En primer lugar, yo no reinventar la rueda ... el método max ya está definido en RichInt, para que pueda escribir a max b, por a y b enteros.

Además, slice es obsoleto, por lo tanto, en lugar de lst.slice(0, num) me gustaría utilizar lst.take(num). Los métodos obsoletos probablemente desaparecerán cuando se inicie Scala 2.8.

EDIT: De hecho, como señaló Daniel, slice(Int, Int) no está obsoleto. Tenía mucha prisa cuando inicialmente escribí esto, y estaba pensando en slice(Int), que es equivalente a drop(Int). Todavía encuentro que lst.take(num) es más claro que lst.slice(0, num) :).

(nitpick) Su última línea tampoco compila, ya que olvidó agregar Nil al final de su secuencia de contras. 1::2::3::4, terminaría invocando :: en una Int, que no tiene este método. Es por eso que debe agregar Nil hasta el final (invoque :: en Nil).

Además, el algoritmo que haya utilizado no es evidente a primera vista. La forma en que iba a escribir esto es la siguiente:

val numbers = /*"--the string of numbers--"*/.map(_.asDigit).toList 

def sliding[A](xs: List[A], w: Int): List[List[A]] = { 
    for(n <- List.range(0, xs.size - w)) 
    yield xs drop n take w 
} 

def product(xs: List[Int]): Int = (1 /: xs) (_ * _) 

sliding(numbers, 5).map(product).sort(_ > _).head 

Siento que la última línea explica bastante bien lo que se supone que el algoritmo para hacer - tener una ventana deslizante de la lista, calcular el producto en esa ventana deslizante y a continuación, obtenga el máximo de los productos calculados (he implementado la función máxima como sort(_ > _).head por pereza, podría haber hecho algo O (n) en lugar de O (n log (n)) si el rendimiento fue crítico ... todavía se ejecuta menos de un segundo)

Tenga en cuenta que la función deslizante estará en la biblioteca Scala 2.8 (consulte Daniel's post, donde me inspiré para escribir esta definición de deslizamiento).

EDIT: Vaya ...perdón por el /:. Solo me gusta la concisión de esto y el hecho de que el elemento inicial del pliegue viene antes de la lista. Se podría escribir de forma equivalente product como el siguiente, para ser más explícita:

def product(xs: List[Int]): Int = xs.foldLeft(1)(_ * _) 

-- Flaviu Cipcigan

+0

Gracias Flaviu, esa es la respuesta que esperaba. Me tomó 5 minutos entender su solución, principalmente debido a la sintaxis muy concisa, que usa. Todavía no conozco operadores, por lo que algo como '1 /: xs' es completamente críptico a primera vista. –

+0

Me alegro de poder ayudar. He editado la publicación para proporcionar la definición menos críptica del producto;). –

+0

¿Se está desaprobando el 'slice'? No veo ninguna advertencia en 2.8 o 2.7.4. –

1

Esta es la forma en que lo hice. Nada sofisticado. En tu código, estabas tomando la longitud de la lista en cada iteración, lo que es un desperdicio. Acabo de agregar el número de 1s (igual que el número de dígitos consecutivos) al final de la lista, así que no es necesario que verifique la longitud de la lista para terminar el ciclo.

val s = ... // string of digits 
val ds = s.map(_.asDigit).toList 

def findMaxProduct(ds: List[Int], n: Int, max: Int): Int = ds match { 
    case Nil => max 
    case _ :: rest => findMaxProduct(rest, n, Math.max(max, ds.take(n).reduceLeft(_ * _))) 
} 

val n = 5 // number of consecutive digits 
println(findMaxProduct(ds ::: List.make(n, 1), n, -1)) 
1

val cadena = ... // cadena de dígitos

nums val = {str.map _. asDigit}
(0 a nums.size-5) .map {i => nums.slice (i, i + 5) .product} .max

y otro, más eficiente:

(0 a nums.size-5) .foldLeft (-1) {case (r, i) => r max nums.slice (i, i + 5) .product}

BTW: trabaja con scala2 .8

1
val bigNumber = """73167176531330624919225119674426574742355349194934 
96983520312774506326239578318016984801869478851843 
85861560789112949495459501737958331952853208805511 
12540698747158523863050715693290963295227443043557 
66896648950445244523161731856403098711121722383113 
62229893423380308135336276614282806444486645238749 
30358907296290491560440772390713810515859307960866 
70172427121883998797908792274921901699720888093776 
65727333001053367881220235421809751254540594752243 
52584907711670556013604839586446706324415722155397 
53697817977846174064955149290862569321978468622482 
83972241375657056057490261407972968652414535100474 
82166370484403199890008895243450658541227588666881 
16427171479924442928230863465674813919123162824586 
17866458359124566529476545682848912883142607690042 
24219022671055626321111109370544217506941658960408 
07198403850962455444362981230987879927244284909188 
84580156166097919133875499200524063689912560717606 
05886116467109405077541002256983155200055935729725 
71636269561882670428252483600823257530420752963450""".replaceAll("\\s+","") 

def getMax(m: Int, l:List[Seq[Int]]): Int = 
    if (l.head.isEmpty) m else 
    getMax(m max l.foldLeft(1) ((acc, l) => acc * l.head), l map (_ tail)) 

def numDigits(bigNum: String, count: Int) = 
    (1 until count).foldLeft(List(bigNumber map (_ asDigit))) ((l, _) => l.head.tail :: l) 

def solve(bigNum: String, count: Int) = getMax(0, numDigits(bigNum, count)) 

solve(bigNumber, 5) 
+0

Esta solución requiere que la cantidad de dígitos consecutivos sea 5. Quería que mi código fuera más genérico. –

+0

¡Yessir! Compruébalo nuevamente. –

Cuestiones relacionadas