2089. 找出数组排序后的目标下标

给你一个下标从 0 开始的整数数组 nums 以及一个目标元素 target

目标下标 是一个满足 nums[i] == target 的下标 i

nums非递减 顺序排序后,返回由 nums 中目标下标组成的列表。如果不存在目标下标,返回一个 列表。返回的列表必须按 递增 顺序排列。

示例 1:

输入:nums = [1,2,5,2,3], target = 2
输出:[1,2]
解释:排序后,nums 变为 [1,2,2,3,5] 。
满足 nums[i] == 2 的下标是 1 和 2 。

示例 2:

输入:nums = [1,2,5,2,3], target = 3
输出:[3]
解释:排序后,nums 变为 [1,2,2,3,5] 。
满足 nums[i] == 3 的下标是 3 。

示例 3:

输入:nums = [1,2,5,2,3], target = 5
输出:[4]
解释:排序后,nums 变为 [1,2,2,3,5] 。
满足 nums[i] == 5 的下标是 4 。

示例 4:

输入:nums = [1,2,5,2,3], target = 4
输出:[]
解释:nums 中不含值为 4 的元素。

提示:

  • 1 <= nums.length <= 100

  • 1 <= nums[i], target <= 100

解题过程

一般思路

核心思想先排序,再寻找目标元素的下标

具体步骤如下:

  1. 对数组 nums 进行非递减顺序排序。

  2. 遍历排序后的数组,寻找所有等于 target 的元素,并记录下它们的下标。

  3. 返回这些下标组成的列表。

class Solution {
    public List<Integer> targetIndices(int[] nums, int target) {
        List<Integer> resultList = new ArrayList<>();
        Arrays.sort(nums);
        for(int i = 0; i < nums.length; i++){
            if(nums[i] > target){
                break;
            }
            if(nums[i] == target){
                resultList.add(i);
            }
        }
        return resultList;
    }
}

进阶思路

其实没必要进行排序,直接统计之前数量,再加上target数量

class Solution {
    public List<Integer> targetIndices(int[] nums, int target) {
        int pre = 0;
        int size = 0;
        for(int i = 0;i < nums.length; i++){
            int temp = nums[i];
            if(temp < target){
                pre++;
            }
            if(temp == target){
                size++;
            }
        }
        List<Integer> result = new ArrayList<>();
        for(int i = 0; i < size; i++){
            result.add(pre+i);
        }
        return result;
    }
}