目标
给你一个整数数组 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 。
说明:
- 1 <= nums.length, queries.length <= 10^5
- 1 <= queries[i] <= 10^5
- 1 <= nums[i], x <= 10^4
思路
查询给定数字 x
在数组 nums
中第 queries[i]
次出现的下标。
一次遍历记录所有 x
的下标即可。
代码
/**
* @date 2024-12-27 6:40
*/
public class OccurrencesOfElement3159 {
public int[] occurrencesOfElement_v1(int[] nums, int[] queries, int x) {
int n = nums.length;
int max = 0;
for (int i = 0; i < n; i++) {
if (nums[i] == x) {
nums[max++] = i;
}
}
int length = queries.length;
for (int i = 0; i < length; i++) {
queries[i] = queries[i] > max ? -1 : nums[queries[i] - 1];
}
return queries;
}
}