2012年5月12日土曜日

Project Euler-Problem8をgroovyで解いてみる

問題

以下の1000桁の数字から5つの連続する数字を取り出して その積を計算する。そのような積の中で最大のものの値はいくらか

73167176531330624919225119674426574742355349194934
96983520312774506326239578318016984801869478851843
85861560789112949495459501737958331952853208805511
12540698747158523863050715693290963295227443043557
66896648950445244523161731856403098711121722383113
62229893423380308135336276614282806444486645238749
30358907296290491560440772390713810515859307960866
70172427121883998797908792274921901699720888093776
65727333001053367881220235421809751254540594752243
52584907711670556013604839586446706324415722155397
53697817977846174064955149290862569321978468622482
83972241375657056057490261407972968652414535100474
82166370484403199890008895243450658541227588666881
16427171479924442928230863465674813919123162824586
17866458359124566529476545682848912883142607690042
24219022671055626321111109370544217506941658960408
07198403850962455444362981230987879927244284909188
84580156166097919133875499200524063689912560717606
05886116467109405077541002256983155200055935729725
71636269561882670428252483600823257530420752963450

EX 6桁の数123789なら、1*2*3*7*8と2*3*7*8*9の二通りとなり、後者の2*3*7*8*9=3024が最大の積となる。

問題を解いたプログラム

Iteratorで連続する5つの数字の積を返して、その最大値を求めてます。
String.metaClass.next = {int size ->
  def str = delegate.toString().replaceAll(/\r|\n/, "")
  if (size > str.size()) {
    return null
  }
  new Iterator<Integer>() {
    def result = []
    def pos = 0
    boolean hasNext() {
      pos < str.size()
    }
    Integer next() {
      if (result.isEmpty()) {
        (str[0..<size]).each {result << it.toInteger()}
        pos = 5
      } else {
        result.remove(0)
        pos <= str.size() ? result.add(str[pos++].toInteger()) : null
      }
      result.inject(1) {n1, n2 -> n1 * nn}
    }

    void remove() {
      throw new UnsupportedOperationException("unsupported remove.")
    }
  }
}
def input = """
73167176531330624919225119674426574742355349194934
96983520312774506326239578318016984801869478851843
85861560789112949495459501737958331952853208805511
12540698747158523863050715693290963295227443043557
66896648950445244523161731856403098711121722383113
62229893423380308135336276614282806444486645238749
30358907296290491560440772390713810515859307960866
70172427121883998797908792274921901699720888093776
65727333001053367881220235421809751254540594752243
52584907711670556013604839586446706324415722155397
53697817977846174064955149290862569321978468622482
83972241375657056057490261407972968652414535100474
82166370484403199890008895243450658541227588666881
16427171479924442928230863465674813919123162824586
17866458359124566529476545682848912883142607690042
24219022671055626321111109370544217506941658960408
07198403850962455444362981230987879927244284909188
84580156166097919133875499200524063689912560717606
05886116467109405077541002256983155200055935729725
71636269561882670428252483600823257530420752963450
"""

println(input.next(5).inject(0) {max, n -> Math.max(max, n)})