LeetCode-Python-170. 两数之和 III - 数据结构设计

最新推荐文章于 2022-08-12 18:00:27 发布

原创最新推荐文章于 2022-08-12 18:00:27 发布 · 1.1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#哈希表

Leetcode 同时被 2 个专栏收录

821 篇文章

订阅专栏

Python

664 篇文章

订阅专栏

博客介绍设计并实现TwoSum类，该类需支持add和find操作。add用于增加数，find用于寻找内部数据结构中是否存在两数之和等于给定数。示例展示了操作结果，思路是用哈希表记录每个数出现次数，同时要注意特殊情况处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

设计并实现一个 TwoSum 的类，使该类需要支持 add 和 find 的操作。

add 操作 - 对内部数据结构增加一个数。
find 操作 - 寻找内部数据结构中是否存在一对整数，使得两数之和与给定的数相等。

示例 1:

add(1); add(3); add(5);
find(4) -> true
find(7) -> false
示例 2:

add(3); add(1); add(2);
find(3) -> true
find(6) -> false

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum-iii-data-structure-design
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

用哈希表实现。用一个哈希表记录每个数出现的次数， key是数字，val是出现的次数。

注意处理key * 2 = 要找的数时的特殊情况。

class TwoSum(object):

    def __init__(self):
        """
        Initialize your data structure here.
        """
        from collections import defaultdict
        self.dict = defaultdict(int)
        

    def add(self, number):
        """
        Add the number to an internal data structure..
        :type number: int
        :rtype: None
        """
        self.dict[number] += 1

    def find(self, value):
        """
        Find if there exists any pair of numbers which sum is equal to the value.
        :type value: int
        :rtype: bool
        """
        for key, val in self.dict.items():
            if value - key in self.dict:
                if key * 2 == value and self.dict[key] > 1:
                    return True
                elif key * 2 != value:
                    return True
        return False