結果

問題 No.694 square1001 and Permutation 3
ユーザー siman
提出日時 2023-06-27 16:02:28
言語 Ruby
(3.4.1)
結果
AC  
実行時間 2,727 ms / 3,000 ms
コード長 1,205 bytes
コンパイル時間 242 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 94,056 KB
最終ジャッジ日時 2024-07-04 09:02:27
合計ジャッジ時間 17,656 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 13
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #
プレゼンテーションモードにする

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] + 1)
bit.add(self[i], 1)
end
cnt
end
end
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
N = gets.to_i
A = N.times.map { gets.to_i }
B = A.uniq.sort
memo = Hash.new
B.each.with_index(1) do |b, i|
memo[b] = i
end
C = A.map { |a| memo[a] }
counter = C.tally
RUI = [0]
C.uniq.sort.each do |c|
cnt = counter[c]
RUI << RUI.last + cnt
end
# pp C
ans = []
cnt = C.inversion_number
ans << cnt
0.upto(N - 2) do |i|
c = C[i]
small_num = RUI[c] - counter[c]
big_num = N - small_num - counter[c]
cnt += (big_num - small_num)
ans << cnt
end
puts ans
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0