結果

問題 No.1854 Limited Bubble Sort
ユーザー yudedakoyudedako
提出日時 2022-03-14 00:17:47
言語 Scala(Beta)
(3.4.0)
結果
AC  
実行時間 1,104 ms / 2,000 ms
コード長 1,965 bytes
コンパイル時間 11,773 ms
コンパイル使用メモリ 269,760 KB
実行使用メモリ 67,268 KB
最終ジャッジ日時 2024-09-19 08:01:27
合計ジャッジ時間 40,206 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 850 ms
63,908 KB
testcase_01 AC 882 ms
63,780 KB
testcase_02 AC 929 ms
63,792 KB
testcase_03 AC 915 ms
64,072 KB
testcase_04 AC 923 ms
64,032 KB
testcase_05 AC 907 ms
64,108 KB
testcase_06 AC 893 ms
64,216 KB
testcase_07 AC 860 ms
64,308 KB
testcase_08 AC 885 ms
64,244 KB
testcase_09 AC 910 ms
64,000 KB
testcase_10 AC 1,104 ms
64,224 KB
testcase_11 AC 1,069 ms
64,900 KB
testcase_12 AC 973 ms
64,176 KB
testcase_13 AC 1,095 ms
65,516 KB
testcase_14 AC 1,000 ms
64,208 KB
testcase_15 AC 954 ms
64,212 KB
testcase_16 AC 1,082 ms
65,284 KB
testcase_17 AC 1,070 ms
65,404 KB
testcase_18 AC 1,064 ms
66,004 KB
testcase_19 AC 1,084 ms
65,488 KB
testcase_20 AC 1,068 ms
66,288 KB
testcase_21 AC 988 ms
64,820 KB
testcase_22 AC 1,062 ms
67,124 KB
testcase_23 AC 1,078 ms
65,484 KB
testcase_24 AC 1,096 ms
67,268 KB
testcase_25 AC 965 ms
64,568 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.PrintWriter
import scala.collection.mutable.*
import scala.io.StdIn.*
import scala.util.chaining.*
import scala.math.*
import scala.reflect.ClassTag
import scala.util.*
import scala.annotation.tailrec
import scala.collection.mutable


def solve(permutation: Array[Int]): Option[ArrayBuffer[Int]] =
  if permutation.forall(permutation.indices.contains(_)) then
    permutation.indices.find(i => permutation(i) == i) match
      case Some(i) =>
        solve(permutation.slice(0, i)).flatMap(result => solve(permutation.slice(i + 1, permutation.length).map(_ - i - 1)).map(_.map(_ + i + 1) ++ result))
      case None  =>
        Some(solveFlat(permutation))
  else
    None

def solveFlat(array: Array[Int]): ArrayBuffer[Int] =
  if array.isEmpty then
    return ArrayBuffer.empty
  val indices = Array.fill(array.length){0}
  val leftCount = Array.fill(array.length){0}
  for (a, i) <- array.zipWithIndex do
    indices(a) = i
    leftCount(a) = (0 until i).count(j => array(j) < array(i))
  val result = ArrayBuffer[Int]()
  val current = array.clone()
  def swap(i: Int) =
    if current(i) < current(i + 1) then
      leftCount(current(i + 1)) -= 1
    else
      leftCount(current(i)) += 1
    val temp = current(i)
    current(i) = current(i + 1)
    current(i + 1) = temp
    indices(current(i)) = i
    indices(current(i + 1)) = i + 1
    result.addOne(i + 1)
  for i <- array.indices do
    for j <- i until indices(i) do
      if current(j) != j + 1 && current(j + 1) == j + 2 && leftCount(j + 2) + 1 != j + 2 then
        swap(j)
    for j <- indices(i) until i by -1 do
      swap(j - 1)
  result


@main def main =
  val testCount = readLine().toInt
  for testCase <- 1 to testCount do
    val n = readLine().toInt
    val permutation = readLine().split(' ').map(_.toInt - 1)
    solve(permutation) match
      case Some(result) =>
        println(result.size)
        println(result.mkString(" "))
      case None =>
        println(-1)
0