Käännöt

class Array
  def swap(index, size)
    if index < 0 || (index + size) > self.length
      throw 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 4 not in place to 4 in place at once
      # because we can only change parity for 4 elements at once

      #puts "inplace = #{in_place}"
      return needs_swap % 4 == 0
    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 sort_4(array, size)
  index = 0
  max_index = array.length - 1
  ops = []
  while true
    max = array[0..max_index].max
    max_index = array.index(max)
    if max - 1 != max_index
      if (max - max_index - 1) % 3 != 0 && max - max_index > 3 && max_index > 0
        ops << max_index - 1
        array.swap(max_index - 1, size)
      elsif (max - max_index - 1) % 3 != 1 && max - max_index < 3 && max_index > 4
        ops << max_index - 3
        ops << max_index - 4
        array.swap(max_index - 3, size)
        array.swap(max_index - 4, size)
        if (max - max_index - 1) % 3 == 1
          ops << max_index - 3
          array.swap(max_index - 3, size)
        end
      elsif max_index == 0 && (max - max_index - 1) % 3 != 0
        ops << max_index
        array.swap(max_index, size)
        if (max - max_index - 1) % 3 == 2
          ops << max_index + 1
          array.swap(max_index + 1, size)
        else
          ops << max_index + 2
          array.swap(max_index + 2, size)
        end
      end
      max_index = array.index(max)
      while ((max - max_index - 1) % 3 == 0) && (max - 1 != max_index)
        ops << max_index
        array.swap(max_index, size)
        max_index += 3
      end
    end
    break if max_index < size
    max_index -= 1
  end
  #puts array.to_s
  return ops.map { |n| n + 1 }
end


_, 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_4(arr, 4)
    puts "YES"
    puts sorted.length
    puts sorted.join(" ")
  else
    puts "NO"
  end
when 5
  sorted = sort(arr, 5)
end

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED

Test 10

Group: 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED

Test 14

Group: 2

Verdict: ACCEPTED

Test 15

Group: 2

Verdict: ACCEPTED

Test 16

Group: 2

Verdict: ACCEPTED

Test 17

Group: 3

Verdict: ACCEPTED

Test 18

Group: 3

Verdict: RUNTIME ERROR

input/code.rb:4:in `throw': uncaught throw IndexError (UncaughtThrowError)
	from input/cod...

Test 19

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 20

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 21

Group: 3

Verdict: WRONG ANSWER

Test 22

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 23

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 24

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 25

Group: 3

Verdict: ACCEPTED

Test 26

Group: 4

Verdict: WRONG ANSWER

Test 27

Group: 4

Verdict: WRONG ANSWER

Test 28

Group: 4

Verdict: WRONG ANSWER

Test 29

Group: 4

Verdict: WRONG ANSWER

Test 30

Group: 4

Verdict: WRONG ANSWER

Test 31

Group: 4

Verdict: WRONG ANSWER

Test 32

Group: 4

Verdict: WRONG ANSWER

Test 33

Group: 4

Verdict: WRONG ANSWER

Test 34

Group: 4

Verdict: WRONG ANSWER