結果

問題 No.238 Mr. K's Another Gift
ユーザー ともき
提出日時 2015-07-06 04:46:07
言語 Scala(Beta)
(3.6.2)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,257 bytes
コンパイル時間 8,434 ms
コンパイル使用メモリ 258,476 KB
実行使用メモリ 178,236 KB
最終ジャッジ日時 2024-06-29 02:20:32
合計ジャッジ時間 73,025 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 TLE * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

import scala.io.StdIn.readLine
import scala.collection.mutable.PriorityQueue
import scala.annotation.tailrec
import scala.util.control.Breaks

object Lib {
  object String {
    // ex. b = 26 (if alphabet), mod = 1000000007
    class RollingHash[T](val b : Long,val mod : Long,val conv : T => Long) {
      final def hash(n : T) : Long = conv(n)%mod
      // n = (c0,c1,c2,c3,...cn)
      //  => (c0*b^n + ... + cn*b^0) mod 'mod'
      final def hash(n : Traversable[T]) =
        n.foldLeft(0l)((c,t) => (conv(t)+b*c)%mod)
      final def scan(n : Traversable[T]) =
        n.scanLeft(0l)((c,t) => (conv(t)+b*c)%mod)
      final def apply(n : T) = hash(n)
      final def apply(n : Traversable[T]) = hash(n)
    }
    object RollingHash {
      def apply[T](b : Long, mod : Long, conv : T => Long) = new RollingHash(b,mod,conv)
    }
  }
}

object Main {
  def solve(s : String) : Option[String] = {
    val r = Lib.String.RollingHash(103,1000000007,(c : Char) => c-'a'+1)
    val a = r.scan(s).toVector
    val b = r.scan(s.reverse).toVector
    val n = s.length
    val p = (0 to n).scanLeft(1l)((i,j) => (i*r.b % r.mod))
    def insert(hashlist : Vector[Long], c : Char, i : Int) : Long = {
      val left  = (hashlist(i) * p(n-i))       % r.mod
      val right = (hashlist(n) - left + r.mod) % r.mod
      val ins   = (r(c) * p(n-i))              % r.mod
      val here = (r.b * left + ins + right)    % r.mod
      here
    }

    // var ret : Option[String] = None
    // val bk = new Breaks
    // bk.breakable {
    //   for(i <- 0 to n;
    //       c <- 'a' to 'z'){
    //     if(insert(a,c,i) == insert(b,c,n-i)){
    //       ret = Some(s.take(i) + c + s.drop(i))
    //       bk.break
    //     }
    //   }
    // }
    // ret
     (for(i <- 0 to n;
          c <- 'a' to 'z') yield (i,c))
       .find({case (i,c) => insert(a,c,i) == insert(b,c,n-i)})
       .map({case (i,c) => s.take(i) + c + s.drop(i)})
      // .filter({case (i,c) => insert(a,c,i) == insert(b,c,n-i)})
     //  .headOption
     //  .map({case (i,c) => s.take(i) + c + s.drop(i)})
  }

  def main(args : Array[String]) : Unit = {
    val s = readLine
    val ret = solve(s) match {
        case Some(j) => j
        case None    => "NA"
      }
    println(ret)
  }
}
0