leetcode 3159.查询数组中元素的出现位置

1.题目要求:

给你一个整数数组 nums ，一个整数数组 queries 和一个整数 x 。

对于每个查询 queries[i] ，你需要找到 nums 中第 queries[i] 个 x 的位置，并返回它的下标。如果数组中 x 的出现次数少于 queries[i] ，该查询的答案为 -1 。

请你返回一个整数数组 answer ，包含所有查询的答案。

 

示例 1：

输入：nums = [1,3,1,7], queries = [1,3,2,4], x = 1

输出：[0,-1,2,-1]

解释：

第 1 个查询，第一个 1 出现在下标 0 处。
第 2 个查询，nums 中只有两个 1 ，所以答案为 -1 。
第 3 个查询，第二个 1 出现在下标 2 处。
第 4 个查询，nums 中只有两个 1 ，所以答案为 -1 。
示例 2：

输入：nums = [1,2,3], queries = [10], x = 5

输出：[-1]

解释：

第 1 个查询，nums 中没有 5 ，所以答案为 -1 。

2.题目代码:

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* occurrencesOfElement(int* nums, int numsSize, int* queries, int queriesSize, int x, int* returnSize) {
    int* x_index = (int*)malloc(sizeof(int) * numsSize);//记录x的下标的位置
    int j = 0;
    //1.遍历数组，查找x下标
    for(int i = 0;i < numsSize;i++){
        if(nums[i] == x){
            x_index[j] = i;
            j++;
        }
    }
    int* index = (int*)malloc(sizeof(int) * queriesSize);//放入出现位置的下标
    int i = 0;
    //遍历queries数组，查找元素位置
    for(i = 0;i < queriesSize;i++){
        if(queries[i] > j){
            index[i] = -1;
        }else{
            index[i] = x_index[queries[i] - 1];
        }
    }
    *returnSize = queriesSize;
    return index;
}