結果

問題 No.2382 Amidakuji M
ユーザー siman
提出日時 2023-07-19 10:58:51
言語 Ruby
(3.4.1)
結果
AC  
実行時間 508 ms / 2,000 ms
コード長 1,230 bytes
コンパイル時間 69 ms
コンパイル使用メモリ 7,296 KB
実行使用メモリ 26,752 KB
最終ジャッジ日時 2024-09-19 09:37:39
合計ジャッジ時間 6,663 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.rb:59: warning: ambiguous first argument; put parentheses or a space even after `-' operator
Syntax OK

ソースコード

diff #

class BinaryIndexTree
  def initialize(size, init: 0)
    @values = Array.new(size, init)
    @size = size
  end

  # @param idx [Integer]
  # @param x [Numeric]
  def add(idx, x)
    raise 'Out of range reference' if @size <= idx

    idx += 1

    while idx <= @size
      @values[idx - 1] += x
      idx += idx & -idx
    end
  end

  def sum(l, r)
    _sum(r) - _sum(l)
  end

  private

  def _sum(idx)
    res = 0

    while idx > 0
      res += @values[idx - 1]
      idx -= idx & -idx
    end

    res
  end
end

class Array
  def inversion_number
    n = size
    bit = BinaryIndexTree.new(n + 1)
    cnt = 0

    n.times do |i|
      cnt += i - bit.sum(0, self[i])
      bit.add(self[i], 1)
    end

    cnt
  end
end

N, M = gets.split.map(&:to_i)
P = gets.split.map(&:to_i)

num = P.inversion_number

if M.even? && num.odd?
  puts -1
elsif num == 0
  puts 0
else
  if M.even?
    k = (num + M - 1) / M
    puts k * M
  else
    diff = (M - num).abs

    if num <= M
      if diff.even?
        puts M
      else
        puts 2 * M
      end
    else
      k = (diff + M - 1) / M
      n_diff = (k * M - num).abs

      if n_diff.even?
        puts k * M
      else
        puts (k + 1) * M
      end
    end
  end
end
0