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

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

给你一个下标从 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

解题思路

以为题目需要将 nums 按 非递减 顺序排序后，返回由 nums 中目标下标组成的列表，因为返回的是进行排序以后的元素下标，而不需要考虑数组元素原来的下标位置，因此我们可以直接对数组进行一次排序，然后对排序以后的数组进行一次遍历，检查是否满足 nums[i] == target，如果满足则将下标加入结果数组中。

代码

class Solution {
public:
    vector<int> targetIndices(vector<int>& nums, int target) {

        sort(nums.begin(),nums.end());
        vector<int> res;
        for (int i = 0; i < nums.size(); ++i) {
            if (nums[i]==target)
            res.push_back(i);
        }
        return res;
    }
};

优化思路

不需要进行排序，只需要记录比target小的元素，就可以获取到 nums[i] == target，i在数组中的下标

代码

class Solution {
public:
    vector<int> targetIndices(vector<int>& nums, int target) {
        int less(0),e(0);
        for (int i = 0; i < nums.size(); ++i) {
            if (nums[i]==target)
                e++;
            else if (nums[i]<target)
                less++;
        }
        vector<int> res;
        for (int i = 0; i < e; ++i) {
            res.push_back(less+i);
        }
        return res;
    }
};