LeetCode 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 。

最后的结果一定是连续的下标,可以先遍历一遍数组nums,找出有几个数字比目标元素小(即起始下标)和目标数字有几个:

class Solution {
public:
    vector<int> targetIndices(vector<int>& nums, int target) {
        int targetNum = 0;
        int beginIndex = 0;
        for (int i : nums) {
            if (i == target) {
                ++targetNum;
            } else if (i < target) {
                ++beginIndex;
            }
        }

        vector<int> res;
        for (int i = 0; i < targetNum; ++i) {
            res.push_back(beginIndex + i);
        }
        return res;
    }
};

你可能感兴趣的:(LeetCode,leetcode,排序算法,算法)