要求
设计一个接收整数流的数据结构,该数据结构支持检查是否存在两数之和等于特定值。
实现 TwoSum 类:
- TwoSum() 使用空数组初始化 TwoSum 对象
- void add(int number) 向数据结构添加一个数 number
- boolean find(int value) 寻找数据结构中是否存在一对整数,使得两数之和与给定的值相等。如果存在,返回 true ;否则,返回 false 。
示例:
输入:
["TwoSum", "add", "add", "add", "find", "find"]
[[], [1], [3], [5], [4], [7]]
输出:
[null, null, null, null, true, false]
解释:
TwoSum twoSum = new TwoSum();
twoSum.add(1); // [] --> [1]
twoSum.add(3); // [1] --> [1,3]
twoSum.add(5); // [1,3] --> [1,3,5]
twoSum.find(4); // 1 + 3 = 4,返回 true
twoSum.find(7); // 没有两个整数加起来等于 7 ,返回 false
提示:
- -105 <= number <= 105
- -231 <= value <= 231 - 1
- 最多调用 104 次 add 和 find
核心代码
class TwoSum:
def __init__(self):
from collections import defaultdict
self.dict = defaultdict(int)
def add(self, number: int) -> None:
self.dict[number] += 1
def find(self, value: int) -> 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
# Your TwoSum object will be instantiated and called as such:
# obj = TwoSum()
# obj.add(number)
# param_2 = obj.find(value)
解题思路:这是一道数据结构的设计题,设计思路比较简单,在find的时候需要注意下,是不是本身,还是其他值,整体比较简单。