-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathP18.java
55 lines (51 loc) · 1.55 KB
/
P18.java
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
import java.util.*;
import java.io.*;
import java.util.ArrayList;
public class P18 {
public static int merge(int[] arr, int left, int mid, int right) {
int count = 0;
int j = mid + 1;
for (int i = left; i <= mid; i++) {
while (j <= right && arr[i] > (2 * (long) arr[j]))
j++;
count += (j - (mid + 1));
}
int[] arr1 = Arrays.copyOfRange(arr, left, mid + 1);
int[] arr2 = Arrays.copyOfRange(arr, mid + 1, right + 1);
int n1 = arr1.length;
int n2 = arr2.length;
int i = 0;
j = 0;
int k = left;
while (i < n1 && j < n2) {
if (arr1[i] <= arr2[j]) {
arr[k++] = arr1[i++];
} else {
arr[k++] = arr2[j++];
}
}
while (i < n1)
arr[k++] = arr1[i++];
while (j < n2)
arr[k++] = arr2[j++];
return count;
}
public static int mergeSort(int[] arr, int left, int right) {
int count = 0;
if (left < right) {
int mid = (left + right) >> 1;
count += mergeSort(arr, left, mid);
count += mergeSort(arr, mid + 1, right);
count += merge(arr, left, mid, right);
}
return count;
}
public static int reversePairs(ArrayList<Integer> a) {
// Extension of Merge Sort algorithm
int[] arr = new int[a.size()];
int j = 0;
for (int i : a)
arr[j++] = i;
return mergeSort(arr, 0, arr.length - 1);
}
}