Submission #434055

---

Source Code Expand

import scala.io.StdIn.readLine

object Main{
  def calc_score(board : Array[Array[Option[Boolean]]],
                 b     : Array[Array[Int]],
                 c     : Array[Array[Int]]) : (Int,Int) = {
    var chokudai = 0
    var chokuko = 0
    for(i <- 0 to 1;
        j <- 0 to 2){
      val bij = b(i)(j)
      if(board(i)(j).get == board(i+1)(j).get)
        chokudai += bij
      else
        chokuko += bij
    }
    for(i <- 0 to 2;
        j <- 0 to 1){
      val cij = c(i)(j)
      if(board(i)(j).get == board(i)(j+1).get)
        chokudai += cij
      else
        chokuko += cij
    }
    (chokudai,chokuko)
  }

  def is_end(board :  Array[Array[Option[Boolean]]]) = board.forall(_.forall(_.isDefined))

  def turn(board    : Array[Array[Option[Boolean]]],
           b        : Array[Array[Int]],
           c        : Array[Array[Int]],
           chokudai : Boolean) : (Int,Int) = {
    if(is_end(board)) calc_score(board,b,c)
    else {
      // val cand = (0 until 3).flatMap((i) =>
      //              (0 until 3).map((j) => {
      //                                board(i)(j) match {
      //                                  case Some(_) => None
      //                                  case None    => {
      //                                    board(i)(j) = Option(chokudai)
      //                                    val k = turn(board,b,c,!chokudai)
      //                                    board(i)(j) = None
      //                                    Option(k)
      //                                  }
      //                                }
      //                              }).collect({case Some(i) => i}))

      val cand = (for(i <- 0 until 3;
                      j <- 0 until 3)
                  yield(i,j))
        .filter({case (i,j) => board(i)(j).isEmpty})
        .map({case (i,j) => {
               board(i)(j) = Option(chokudai)
               val k = turn(board,b,c,!chokudai)
               board(i)(j) = None
               k
             }})
      // var cand = List[(Int,Int)]()
      // for(i <- 0 until 3;
      //     j <- 0 until 3){
      //   board(i)(j) match {
      //     case Some(_) => None
      //     case None    => {
      //       board(i)(j) = Option(chokudai)
      //       cand = turn(board,b,c,!chokudai) :: cand
      //       board(i)(j) = None
      //     }
      //   }
      // }
      if(chokudai)
        cand.maxBy(_._1)
      else
        cand.maxBy(_._2)
    }
  }

  def main(args : Array[String]) : Unit = {
    val board = Array.fill(3)(Array.fill(3)(Option.empty[Boolean]))
    val b = (1 to 2).map((i) => readLine().split(" ").map(_.toInt)).toArray
    val c = (1 to 3).map((i) => readLine().split(" ").map(_.toInt)).toArray
    val last = turn(board,b,c,true)
    println(last._1)
    println(last._2)
  }
}

Submission Info

Judge Result