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

Posted 2022-12-13 潜行前行

题目地址(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++;