题目地址(315. 计算右侧小于当前元素的个数)-CSDN博客

本文链接：https://blog.csdn.net/u013591094/article/details/124933651

该博客介绍了如何利用归并排序算法解决一个数组中每个元素右侧有多少更小元素的问题。通过维护一个额外的计数数组和原始元素的索引，博主详细解析了算法的实现过程，包括归并排序的分治思想和合并过程中如何更新计数。示例代码以Java编写，展示了具体的操作步骤和逻辑。

摘要由CSDN通过智能技术生成

题目地址(315. 计算右侧小于当前元素的个数)

https://leetcode.cn/problems/count-of-smaller-numbers-after-self/

题目描述

给你一个整数数组 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 <= nums.length <= 105
-104 <= nums[i] <= 104

关键点

归并排序+记录原始的index

代码

语言支持：Java

Java Code:


class Solution {
    int[] index; // 记录原始的位置
    int[] count; // 需要累计
    int[] tempIndex;
    int[] temp;
    public List<Integer> countSmaller(int[] nums) {
        index = new int[nums.length];
        for (int i = 0; i < nums.length; i++) {
            index[i] = i;
        }
        count = new int[nums.length];
        temp = new int[nums.length]; // 提前生成
        tempIndex = new int[nums.length]; // 提前生成

        mergeSort(nums,0,nums.length-1);
        List<Integer> ans = new ArrayList<>();
        for(int i=0;i<nums.length;i++){
            ans.add(count[i]);
        }
        return ans;
    }

    void mergeSort(int[] nums,int left ,int right){
        if(left == right) return;
        int mid = left + ((right-left) >> 1);
        mergeSort(nums,left,mid);
        mergeSort(nums,mid+1,right);
        if(nums[mid] > nums[mid+1]){ // 右侧 大于 左右 不需再合并。已经得到结果了
            merge(nums,left,right,mid);
        }
    }
    void merge(int[] nums,int left ,int right,int mid){
        int i = left;
        int j = mid+1;
        for (int x = left; x <= right; x++) {
            temp[x] = nums[x];
            tempIndex[x] = index[x]; //最原始的数组下标
        }
        for (int k = left; k <= right; k++) { 
            if((j == right+1 || temp[i] <= temp[j]) && i < mid+1  ){
                nums[k] = temp[i];
                index[k] = tempIndex[i]; 
                count[tempIndex[i]] += (j-mid-1);
                i++;
            }else{ // temp[i] > temp[j] || i == mid +1
                nums[k] = temp[j];
                index[k] = tempIndex[j]; 
                j++;
            }
        }
    }

}