-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathInversionCount.rb
More file actions
37 lines (30 loc) · 819 Bytes
/
InversionCount.rb
File metadata and controls
37 lines (30 loc) · 819 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
module MergeSort
attr_reader :inversion_count
def sort(arr)
@inversion_count = 0
merge_sort(arr)
end
private
def merge_sort(arr)
len = arr.length
return arr if len <= 1
left = merge_sort(arr[0...len/2])
right = merge_sort(arr[len/2...len])
merge(left, right)
end
def merge(left, right)
result = []
until (left.empty? || right.empty?)
result << if left.first <= right.first
left.shift
else
@inversion_count += left.length
right.shift
end
end
result.concat(left).concat(right)
end
end
# include MergeSort
# sort IO.read("input.txt").scan(/\d+/).map!(&:to_i)
# inversion_count