JAVA学习-练习试用Java实现“计算右侧小于当前元素的个数”

问题：

给定一个整数数组 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

解答思路：

题目分析：
- 给定一个整数数组'nums'，要求返回一个新数组'counts'，其中'counts[i]'表示`nums[i]'右侧小于'nums[i]'的元素的数量。
- 示例 1 中，'nums'为'[5,2,6,1]'，对应的'counts'为'[2,1,1,0]'，即 5 右侧有 2 个更小的元素（2 和 1），2 右侧有 1 个更小的元素（1），6 右侧有 1 个更小的元素（1），1 右侧有 0 个更小的元素。
- 示例 2 和示例 3 中，'nums'分别为'[-1]'和'[-1,-1]'，对应的'counts'均为'[0]'。

主要思路：
- 可以使用归并排序的思想来解决这个问题。在归并过程中，同时统计每个元素右侧小于它的元素个数。
- 具体来说，将数组分成两半，分别对左半部分和右半部分进行归并排序，并在归并过程中统计每个元素右侧小于它的元素个数。
- 最后，将左右两半部分合并成一个有序的数组，并将统计的结果存储在'counts'数组中。

以下是修改后的代码实现：

class Solution {
    // 合并两个已排序的子数组，并统计右侧小于当前元素的个数
    private void merge(int[] nums, int[] counts, int[] temp, int left, int mid, int right) {
        int i = left, j = mid + 1, k = left;
        // 统计左侧子数组中右侧小于当前元素的个数
        while (i <= mid) {
            while (j <= right && nums[i] > nums[j]) {
                j++;
            }
            counts[i] += j - mid - 1;
            i++;
        }
        // 合并两个子数组
        i = left;
        j = mid + 1;
        while (i <= mid && j <= right) {
            if (nums[i] <= nums[j]) {
                temp[k++] = nums[i++];
            } else {
                temp[k++] = nums[j++];
            }
        }
        while (i <= mid) {
            temp[k++] = nums[i++];
        }
        while (j <= right) {
            temp[k++] = nums[j++];
        }
        // 将合并后的结果复制回原数组
        System.arraycopy(temp, left, nums, left, right - left + 1);
    }

    // 归并排序函数
    private void mergeSort(int[] nums, int[] counts, int[] temp, int left, int right) {
        if (left < right) {
            int mid = left + (right - left) / 2;
            mergeSort(nums, counts, temp, left, mid);
            mergeSort(nums, counts, temp, mid + 1, right);
            merge(nums, counts, temp, left, mid, right);
        }
    }

    // 计算右侧小于当前元素的个数
    public int[] countSmaller(int[] nums) {
        int n = nums.length;
        int[] counts = new int[n];
        int[] temp = new int[n];
        mergeSort(nums, counts, temp, 0, n - 1);
        return counts;
    }

    // 测试用例
    public static void main(String[] args) {
        int[] nums = {5, 2, 6, 1};
        Solution solution = new Solution();
        int[] counts = solution.countSmaller(nums);
        for (int count : counts) {
            System.out.print(count + " ");
        }
    }
}

在上述代码中，'merge'函数用于合并两个已排序的子数组，并统计右侧小于当前元素的个数。'mergeSort'函数用于对数组进行归并排序。'countSmaller'函数用于计算右侧小于当前元素的个数。

在'main'方法中，创建了一个示例数组'nums'，调用'countSmaller'函数计算右侧小于当前元素的个数，并打印结果。

(文章为作者在学习java过程中的一些个人体会总结和借鉴，如有不当、错误的地方，请各位大佬批评指正，定当努力改正，如有侵权请联系作者删帖。)