Uolevin kalansaalis

class Array
  def swap(index, size)
    if index < 0 || (index + size) > self.length
      raise IndexError
    end
    self[index...(index+size)] = self[index...(index+size)].reverse
  end

  def is_possible(size)
    case size
    when 3
      self.each_with_index do |val, ind|
        if (val + ind) % 2 == 0
          return false
        end
      end
      return true
    when 4
      needs_swap = 0
      self.each_with_index do |val, ind|
        if (val + ind) % 2 == 0
          needs_swap += 1
        end
      end
      # we can only make 2 not in place to 2 in place at once
      # because we can only change parity for 2 elements at once

      #puts "inplace = #{in_place}"
      return needs_swap# % 2 == 0
    when 5
      return self.is_possible(3)

=begin
      total = 0
      self.each_with_index do |val, ind|
        total += (val - 1) - ind
      end
      return total == 0
=end
    end
  end
end

def sort(array, size)
  continue = true
  index = 0
  max_index = array.length - size + 1
  ops = []
  while continue
    continue = false
    while index < max_index
      if array[index] > array[index + size - 1]
        ops << index + 1 # shift index for the "correct index"
        array.swap(index, size)
        continue = true
      end
      index += 1
    end
    index = 0
    max_index -= 1
  end
  return ops
end

def moves_needed_5(arr)
  newarr = []
  arr.each_with_index do |val, ind|
    newarr << ((ind + 1) - val).abs
  end
  newarr
end

input = ""
array = [1,4,3,6,5,2,7,8,9]#[3, 4, 1, 2, 5, 8, 7, 6, 9]
while true
  puts("input:")
  input = gets.chomp[0].to_i
  array.swap(input, 5)
  p array
  mn = moves_needed_5(array)
  p mn
  a1 = 0
  a2 = 0
  mn.each_with_index do |v, i|
    if i % 2 == 0
      a1 += v
    else
      a2 += v
    end
  end
  p a1
  p a2
  puts array.is_possible(5)
end
exit()

# for 5, create array with how many indexes need to move?
# ex: [3, 4, 1, 2, 5, 8, 7, 6, 9]
# ==> [2, 2,-2,-2, 0, 2, 0, -2, 0]
_, swap = gets.chomp.split(" ").map(&:to_i)
arr = gets.chomp.split(" ").map(&:to_i)

case swap
when 2
  puts "YES"
  sorted = sort(arr, 2)
  puts sorted.length
  puts sorted.join(" ")
when 3
  if arr.is_possible(3)
    puts "YES"
    sorted = sort(arr, 3)
    puts sorted.length
    puts sorted.join(" ")
  else
    puts "NO"
  end
when 4
  if arr.is_possible(4)
    sorted = sort(arr, 4)
    puts "YES"
    puts sorted.length
    puts sorted.join(" ")
  else
    puts "NO"
  end
when 5
  if arr.is_possible(3)
    puts "YES"
    sorted = sort(arr, 5)
    puts sorted.length
    puts sorted.join(" ")
  else
    puts "NO"
  end
end

Test 1

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 2

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 3

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 4

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 5

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 6

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 7

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 8

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 9

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:77:in `<main>': undefined method `chomp' for nil:NilClass (NoMethodError)

Test 10

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:77:in `<main>': undefined method `chomp' for nil:NilClass (NoMethodError)

Test 11

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:77:in `<main>': undefined method `chomp' for nil:NilClass (NoMethodError)

Test 12

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 13

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 14

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 15

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 16

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 17

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 18

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 19

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 20

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 21

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 22

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 23

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'

Test 24

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:4:in `swap': IndexError (IndexError)
	from input/code.rb:78:in `<main>'