結果

問題 No.429 CupShuffle
ユーザー gemmarogemmaro
提出日時 2020-07-19 09:06:29
言語 Ruby
(3.3.0)
結果
AC  
実行時間 365 ms / 2,000 ms
コード長 1,014 bytes
コンパイル時間 105 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 45,312 KB
最終ジャッジ日時 2024-05-09 15:21:17
合計ジャッジ時間 3,697 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 76 ms
12,416 KB
testcase_01 AC 71 ms
12,160 KB
testcase_02 AC 71 ms
12,288 KB
testcase_03 AC 70 ms
12,160 KB
testcase_04 AC 76 ms
12,288 KB
testcase_05 AC 78 ms
12,032 KB
testcase_06 AC 79 ms
12,288 KB
testcase_07 AC 91 ms
12,288 KB
testcase_08 AC 72 ms
12,160 KB
testcase_09 AC 77 ms
12,288 KB
testcase_10 AC 102 ms
14,848 KB
testcase_11 AC 360 ms
43,136 KB
testcase_12 AC 365 ms
45,056 KB
testcase_13 AC 349 ms
45,312 KB
testcase_14 AC 356 ms
44,800 KB
testcase_15 AC 73 ms
12,160 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

# frozen_string_literal: true

class Array
  def swap!(a, b)
    ia = a
    ib = b
    self[ia], self[ib] = self[ib], self[ia]
  end

  def divide_by_question
    question_index = index { _1[0] == '?' }
    if question_index.zero?
      [[], to_integer]
    else
      [self[0..(question_index - 1)], self[(question_index + 1)..-1]]
        .map { _1.to_integer }
    end
  end

  def to_integer
    map { _1.map(&:to_i) }
  end

  def move_cups(moves)
    moves.each do |left, right|
      swap!(left - 1, right - 1)
    end

    self
  end

  def detect_position(other)
    zip(other)
      .map
      .with_index { |pair, index| [pair[0] != pair[1], index] }
      .filter { |flag, _index| flag }
      .map { |_flag, index| index + 1 }
      .join(' ')
  end
end

def solve
  t, b = A.divide_by_question
  ct = (1..N).to_a.move_cups(t)
  cb = C.move_cups(b.reverse)
  ct.detect_position(cb)
end

N, K, X = gets.split.map(&:to_i)
A = K.times.map { gets.chomp.split }
C = gets.chomp.split.map(&:to_i)

puts solve
0