给你一个整数数组 nums ,按要求返回一个新数组 counts 。数组 counts 有该性质: counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。
示例 1:
输入:nums = [5,2,6,1]
输出:[2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1)
2 的右侧仅有 1 个更小的元素 (1)
6 的右侧有 1 个更小的元素 (1)
1 的右侧有 0 个更小的元素
示例 2:
输入:nums = [-1]
输出:[0]
示例 3:
输入:nums = [-1,-1]
输出:[0,0]
思路1:归并
public class Solution {
int cnt;
public int InversePairs(int[] array) {
cnt = 0;
if (array != null)
mergeSortUp2Down(array, 0, array.length - 1);
return cnt;
}
/*
* 归并排序(从上往下)
*/
public void mergeSortUp2Down(int[] a, int start, int end) {
if (start >= end)
return;
int mid = (start + end) >> 1;
mergeSortUp2Down(a, start, mid);
mergeSortUp2Down(a, mid + 1, end);
merge(a, start, mid, end);
}
/*
* 将一个数组中的两个相邻有序区间合并成一个
*/
public void merge(int[] a, int start, int mid, int end) {
int[] tmp = new int[end - start + 1];
int i = start, j = mid + 1, k = 0;
while (i <= mid && j <= end) {
if (a[i] <= a[j])
tmp[k++] = a[i++];
else {
tmp[k++] = a[j++];
cnt += mid - i + 1; // core code, calculate InversePairs............
//cnt = cnt % 1000000007; 100%
}
}
while (i <= mid)
tmp[k++] = a[i++];
while (j <= end)
tmp[k++] = a[j++];
for (k = 0; k < tmp.length; k++)
a[start + k] = tmp[k];
}
}
思路2 : sortList
class Solution {
public List<Integer> countSmaller(int[] arr) {
List<Integer> list = new ArrayList<>(arr.length);
if (arr.length == 0) {
return list;
}
List<Integer> sortList = new ArrayList<>(arr.length);
for (int i = arr.length - 1; i >= 0; i--) {
sortList.add(arr[i]);
Collections.sort(sortList);
int moveCount = sortList.indexOf(arr[i]);
list.add(moveCount);
}
Collections.reverse(list);
return list;
}
}
[5,2,6,1]
1:[1],排序后[1],1的index = 0
6:[1,6],排序后[1,6],6的index = 1
2:[1,6,2],排序后[1,2,6],2的index = 1
5:[1,2,6],排序后[1,2,5,6],5的index = 2
所以[5,2,6,1]对应的逆向对:[2,1,1,0]