結果

問題 No.1779 Magical Swap
ユーザー yudedakoyudedako
提出日時 2022-02-02 13:28:25
言語 Scala(Beta)
(3.4.0)
結果
AC  
実行時間 1,376 ms / 2,000 ms
コード長 1,208 bytes
コンパイル時間 10,280 ms
コンパイル使用メモリ 261,440 KB
実行使用メモリ 81,560 KB
最終ジャッジ日時 2024-06-11 09:21:26
合計ジャッジ時間 32,986 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 838 ms
63,820 KB
testcase_01 AC 860 ms
63,884 KB
testcase_02 AC 1,110 ms
75,828 KB
testcase_03 AC 872 ms
64,124 KB
testcase_04 AC 1,376 ms
80,916 KB
testcase_05 AC 1,117 ms
68,168 KB
testcase_06 AC 1,142 ms
69,492 KB
testcase_07 AC 1,310 ms
76,812 KB
testcase_08 AC 857 ms
64,044 KB
testcase_09 AC 1,055 ms
65,716 KB
testcase_10 AC 1,324 ms
81,436 KB
testcase_11 AC 1,102 ms
70,876 KB
testcase_12 AC 1,080 ms
65,872 KB
testcase_13 AC 1,303 ms
74,396 KB
testcase_14 AC 1,347 ms
81,448 KB
testcase_15 AC 1,144 ms
74,736 KB
testcase_16 AC 1,267 ms
81,560 KB
testcase_17 AC 1,215 ms
65,660 KB
testcase_18 AC 834 ms
63,696 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import scala.collection.mutable
import scala.collection.mutable.ArrayBuffer
import scala.io.StdIn.*
import scala.util.chaining.*
import scala.math.*

def calcPrime(upperLimit: Int): Array[Int] =
  val isPrime = Array.fill(upperLimit + 1){true}
  for p <- 2 to sqrt(upperLimit).toInt do
    if isPrime(p) then
      for j <- p * p to upperLimit by p do
        isPrime(j) = false
  (2 to upperLimit).filter(isPrime).toArray

extension (array: Array[Int])
  def upperBound(value: Int): Int =
    var min = 0
    var max = array.length
    while min < max do
      val mid = (min + max) >>> 1
      if array(mid) <= value then
        min = mid + 1
      else
        max = mid
    max

@main def main =
  val prime = calcPrime(100000)
  val testCase = readLine().toInt
  val isOk = Array.fill(testCase){false}
  for t <- 0 until testCase do
    val n = readLine().toInt
    val a = readLine().split(' ').map(_.toInt)
    val b = readLine().split(' ').map(_.toInt)
    if a.sorted sameElements b.sorted then
      isOk(t) = a(0) == b(0) && (prime.upperBound(n / 2) until prime.upperBound(n)).forall(i => a(prime(i) - 1) == b(prime(i) - 1))
  println(
    isOk.map(if _ then "Yes" else "No").mkString("\n")
  )
0