3159. 查询数组中元素的出现位置
给你一个整数数组
nums
,一个整数数组 queries
和一个整数 x
。对于每个查询 queries[i]
,你需要找到 nums
中第 queries[i]
个 x
的位置,并返回它的下标。如果数组中 x
的出现次数少于 queries[i]
,该查询的答案为 -1 。
请你返回一个整数数组 answer
,包含所有查询的答案。
class Solution: def occurrencesOfElement(self, nums: List[int], queries: List[int], x: int) -> List[int]: ans = [] for i,n in enumerate(nums): if n==x: ans.append(i) res = [] for q in queries: if q>len(ans): res.append(-1) else: res.append(ans[q-1]) return res