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;}
};