二分算法：练习题目15

题目：距离最小相等元素查询

给你一个 循环 数组 nums 和一个数组 queries 。

对于每个查询 i ，你需要找到以下内容：

• 数组 nums 中下标 queries[i] 处的元素与 任意 其他下标 j（满足 nums[j] == nums[queries[i]]）之间的 最小 距离。如果不存在这样的下标 j，则该查询的结果为 -1 。

返回一个数组 answer，其大小与 queries 相同，其中 answer[i] 表示查询i的结果。

示例 1：

输入： nums = [1,3,1,4,1,3,2], queries = [0,3,5]

输出： [2,-1,3]

解释：

• 查询 0：下标 queries[0] = 0 处的元素为 nums[0] = 1 。最近的相同值下标为 2，距离为 2。
• 查询 1：下标 queries[1] = 3 处的元素为 nums[3] = 4 。不存在其他包含值 4 的下标，因此结果为 -1。
• 查询 2：下标 queries[2] = 5 处的元素为 nums[5] = 3 。最近的相同值下标为 1，距离为 3（沿着循环路径：5 -> 6 -> 0 -> 1）。

示例 2：

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

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

解释：

数组 nums 中的每个值都是唯一的，因此没有下标与查询的元素值相同。所有查询的结果均为 -1。

提示：

• 1 <= queries.length <= nums.length <= 105
• 1 <= nums[i] <= 106
• 0 <= queries[i] < nums.length

---

参考答案：

from collections import defaultdict
from bisect import bisect_left
from bisect import bisect_right
class Solution:
    def solveQueries(self, nums: List[int], queries: List[int]) -> List[int]:
        pos = defaultdict(list)
        for i,n in enumerate(nums):
            pos[n].append(i)
        lenth = len(nums)
        def select(pos,queries,lenth):
            ans = []
            # 这里疯狂报错，原来把listt的更改写到下一个for里面了，但是insert和append会累加，所以出错，更改循环就没问题了
            for listt in pos.values():
                l0 = listt[0]
                # 注意这里只补一个下标就行不需要补全部，因为这是某一个元素的所有索引值，最近的只可能是相邻的两个
                listt.insert(0,listt[-1]-lenth)
                listt.append(l0+lenth)
            for qi,q in enumerate(queries):
                listt = pos[nums[q]]
                if len(listt) == 3:
                    ans.append(-1)
                else:
                    #因为一定listt中包含这个q，因此用bisectleft就可以得到第一个大于等于q的索引，实际上就是得到等于q的索引，最近的我们只需要从这个索引的左右邻居找就可以了
                    i = bisect_left(listt,q)
                    x = min(q - listt[i-1],listt[i+1] - q)
                    ans.append(x)
            return ans
            
        ans = select(pos,queries,lenth)
        return ans

 真的写了好久，特别是最后一直有部分测试用例不通过，找了很久问题在哪，原来是把listt放在for qi,q in enumerate(queries):这个循环里面了，但是我使用的listt.append还有insert都是在列表本身做修改的，每次循环都会使得listt左右各补一个元素，实际上只需要补一次就行了，这个很重要。