1. 两数之和
知识点:哈希表
正确答案
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
size = len(nums)
num_dict = {}
for i in range(size):
if (target - nums[i]) in num_dict:
return [num_dict[target - nums[i]], i]
num_dict[nums[i]] = i
return []
自己的错误点:
(1) if target - nums[i] in target:
(2) return [[target - nums[i], i]
问题在于对于哈希表还是不熟,if
循环内是通过新生成的哈希表num_dict
判断+返回配对数的下标。
771. 宝石与石头
直接遍历做时间复杂度是O(mn)
,可通过哈希表降低时间复杂度,这里有个知识点:
- 列表数据有序,可重复,查找某个元素方式为逐个遍历,时间复杂度是
O(len(list))
- 集合数据无序,不可重复,查找某个元素方式为哈希,时间复杂度为
O(1)
。对于大量数据,集合效率极高。
哈希解法:
class Solution:
def numJewelsInStones(self, J: str, S: str) -> int:
# 用集合存储 J 中的字符
jewel_set = set()
for j in J:
jewel_set.add(j)
count = 0
# 遍历 S
for s in S:
# 判断是否存在于集合中
if s in jewel_set:
count += 1
return count