※315. 计算右侧小于当前元素的个数(java)(归并排序,分治思想)

给定一个整数数组 nums,按要求返回一个新数组 counts。数组 counts 有该性质: counts[i] 的值是  nums[i] 右侧小于 nums[i] 的元素的数量。

示例:

输入: [5,2,6,1]
输出: [2,1,1,0] 
解释:
5 的右侧有 2 个更小的元素 (2 和 1).
2 的右侧仅有 1 个更小的元素 (1).
6 的右侧有 1 个更小的元素 (1).
1 的右侧有 0 个更小的元素.

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-of-smaller-numbers-after-self
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
     
    int[] count; // 记录最后的结果
    int[] temp; //辅助数组
    int[] indexes; //记录最初的位置     
    public List<Integer> countSmaller(int[] nums) {
     
        ArrayList<Integer> res = new ArrayList<>();
        int len = nums.length;
        if(len == 0) return res;
        temp = new int[len];
        indexes = new int[len];
        count = new int[len];
        for(int i = 0; i < len; i++) {
      
            indexes[i] = i;
        }
        merge(nums, 0, len-1);
        for(int i = 0; i < len; i++) res.add(count[i]);
        return res;
    }
    public void merge(int[] nums, int l, int r) {
     
        if(l == r) return;
        int mid = l + (r-l)/2;
        merge(nums, l, mid);
        merge(nums, mid+1, r);
        if(nums[indexes[mid]] > nums[indexes[mid+1]]) count(nums, l, mid, r); 
    }
    //左边出列,统计逆序数
    public void count(int[] nums, int l, int mid, int r) {
     
        for(int i = l; i <= r; i++) temp[i] = indexes[i];
        int i = l; 
        int j = mid+1;
        for(int k = l; k <= r; k++) {
     
            if(i > mid) {
     
                indexes[k] = temp[j];
                j++;
            } else if(j > r) {
     
                indexes[k] = temp[i];
                i++;
                count[indexes[k]] += (r-mid);
            } else if(nums[temp[i]] <= nums[temp[j]]) {
     
                indexes[k] = temp[i];
                i++;
                count[indexes[k]] += (j-mid-1);
            } else {
     
                indexes[k] = temp[j];
                j++;
            }
        }
    }
}

你可能感兴趣的:(leetcode,java(困难),leetcode,算法,数据结构)