結果

問題 No.238 Mr. K's Another Gift
ユーザー ともきともき
提出日時 2015-07-06 04:34:14
言語 Scala(Beta)
(3.4.0)
結果
AC  
実行時間 1,843 ms / 2,000 ms
コード長 3,333 bytes
コンパイル時間 10,086 ms
コンパイル使用メモリ 273,836 KB
実行使用メモリ 88,348 KB
最終ジャッジ日時 2024-06-29 02:17:43
合計ジャッジ時間 53,270 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 861 ms
67,048 KB
testcase_01 AC 874 ms
65,400 KB
testcase_02 AC 903 ms
65,276 KB
testcase_03 AC 903 ms
65,880 KB
testcase_04 AC 989 ms
66,492 KB
testcase_05 AC 1,563 ms
76,388 KB
testcase_06 AC 1,574 ms
85,288 KB
testcase_07 AC 1,701 ms
85,148 KB
testcase_08 AC 1,717 ms
85,280 KB
testcase_09 AC 1,621 ms
85,120 KB
testcase_10 AC 890 ms
65,188 KB
testcase_11 AC 892 ms
65,260 KB
testcase_12 AC 886 ms
65,420 KB
testcase_13 AC 983 ms
66,108 KB
testcase_14 AC 1,403 ms
71,792 KB
testcase_15 AC 1,723 ms
88,180 KB
testcase_16 AC 1,744 ms
88,288 KB
testcase_17 AC 1,769 ms
88,260 KB
testcase_18 AC 1,699 ms
88,348 KB
testcase_19 AC 1,748 ms
88,240 KB
testcase_20 AC 885 ms
65,412 KB
testcase_21 AC 877 ms
65,368 KB
testcase_22 AC 891 ms
65,456 KB
testcase_23 AC 896 ms
65,588 KB
testcase_24 AC 1,004 ms
66,372 KB
testcase_25 AC 978 ms
66,860 KB
testcase_26 AC 1,539 ms
86,184 KB
testcase_27 AC 1,568 ms
85,996 KB
testcase_28 AC 1,010 ms
75,648 KB
testcase_29 AC 1,588 ms
86,216 KB
testcase_30 AC 875 ms
67,380 KB
testcase_31 AC 854 ms
67,124 KB
testcase_32 AC 907 ms
67,232 KB
testcase_33 AC 1,024 ms
67,476 KB
testcase_34 AC 1,587 ms
78,208 KB
testcase_35 AC 1,749 ms
88,148 KB
testcase_36 AC 1,762 ms
87,932 KB
testcase_37 AC 1,740 ms
88,244 KB
testcase_38 AC 1,704 ms
88,252 KB
testcase_39 AC 1,843 ms
88,076 KB
testcase_40 AC 851 ms
67,072 KB
testcase_41 AC 861 ms
67,232 KB
testcase_42 AC 860 ms
67,332 KB
権限があれば一括ダウンロードができます

ソースコード

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 insert(s : String, c : Char, n : Int) : String = {
  //   val (head,tail) = s.splitAt(n)
  //   head + c + tail
  // }
  // def solve(s : String) : Option[String] = {
  //   val n = s.length
  //   val b = new Breaks
  //   def check(s : String,c : Char) : Boolean = {
  //     val n = s.length
  //     (0 until s.length).forall((i) => {
  //                                 val j = n - i - 1
  //                                 s(i) == c || s(j) == c || s(i) == s(j)
  //                               })
  //   }
  //   if(check(s,'?')){
  //     Some(s.take(n/2) + s(n/2) + s.drop(n/2))
  //   }else{
  //     var ans : Option[String] = None
  //     b.breakable {
  //       for(i <- 0 until n){
  //         val j = n - i - 1
  //         if(s(i) != s(j)){
  //           val checking = Vector((s.take(i) + "?" + s.drop(i)),
  //                                 (s.take(j+1) + "?" + s.drop(j+1)))
  //           checking.foreach((i) => if(check(i,'?')){
  //                              val r = i.indexOf('?')
  //                              ans = Some(i.updated(r,i(n-r)))
  //                            })
  //           b.break
  //         }
  //       }
  //     }
  //     ans
  //   }
  // }
  // rsexr
  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))
     //  .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