字典的使用以及深拷贝

题目描述

请你设计一个数据结构，它能求出给定子数组内一个给定值的 频率 。

子数组中一个值的 频率 指的是这个子数组中这个值的出现次数。

请你实现 RangeFreqQuery 类：

RangeFreqQuery(int[] arr) 用下标从 0 开始的整数数组 arr 构造一个类的实例。
int query(int left, int right, int value) 返回子数组 arr[left…right] 中 value 的 频率 。
一个 子数组 指的是数组中一段连续的元素。arr[left…right] 指的是 nums 中包含下标 left 和 right 在内 的中间一段连续元素。

第一想法

使用字典存储从开始元素到对应下标所有的元素及其对应的次数，最后形成一个以字典为元素的数组，最终查询时只需要得到右侧和左侧的字典中该元素的值即可。

注意点：

• 字典的默认值last = tmp.get(arr[i], 0)，不存在的键不能直接操作，需要判断一下
• 深拷贝，否则使用的是同一片空间，不是一个字典数组，仅仅是一个字典的元素
• 如果左侧元素=val需要最终结果加1

代码

    def __init__(self, arr: List[int]):
        self.dic = []
        self.arr = arr
        tmp = {}
        for i in range(len(arr)):
            last = tmp.get(arr[i], 0)
            # 不使用深拷贝
            tmpNew = copy.deepcopy(tmp)
            tmpNew[arr[i]] = last + 1
            tmp = tmpNew
            self.dic.append(tmpNew)

    def query(self, left: int, right: int, value: int) -> int:
        lDic = self.dic[left]
        rDic = self.dic[right]
        rValue = rDic.get(value, 0)
        lValue = lDic.get(value, 0)
        if self.arr[left] == value:
            return rValue - lValue + 1
        return rValue - lValue

结果超时了。

官方解法

字典存储的键也是val，但是值是其出现的下标
self.occurence = defaultdict(list)
# 顺序遍历数组初始化哈希表
n = len(arr)
for i in range(n):
self.occurence[arr[i]].append(i)

比如元素3出现的下标是[2,5,7,8]，字典对应的值存储该列表
通过两次二分查找寻找到数组中第一个大于等于left 的位置 l 与第一个大于right 的位置 r，此时 r−l 即为符合要求的下标个数

    def __init__(self, arr: List[int]):
        # 数值为键，出现下标数组为值的哈希表
        self.occurence = defaultdict(list)
        # 顺序遍历数组初始化哈希表
        n = len(arr)
        for i in range(n):
            self.occurence[arr[i]].append(i)

    def query(self, left: int, right: int, value: int) -> int:
        # 查找对应的出现下标数组，不存在则为空
        pos = self.occurence[value]
        # 两次二分查找计算子数组内出现次数
        l = bisect_left(pos, left)
        r = bisect_right(pos, right)
        return r - l