Leetcode[1][202] python实现 哈希表
Leetcode (1) 两数之和
题目
给定一个整数数组 [nums] 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。 你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
分析
法一:
利用python中的字典(dict)来构建hashtable,与此前数构课程使用的C++有所不同。
初始化一个字典后,利用enumerate()将给定的nums[ ] 使得列表中的每个元素 v 有与之顺序对应的 i ,将其全部存入字典中,其中 i 为value, v 为key。
遍历字典,找到与当前循环的key相加等于target的数,返回对应的[i1, i2]。
法2:
只进行了1次循环。与法1区别在于构建哈希表时并不是直接存入对应的 i 和 v ,而是存入 target-v 和相对应的 i ,同时边构建边判断 v 是否已在字典中,若已在,则可直接返回结果。
代码
class Solution:
def twoSum1(self, nums, target): #48ms
hashtable = {}
for i,v in enumerate(nums):
hashtable[v] = i
for i2,v2 in enumerate(nums):
if target-v2 in hashtable and i2 != hashtable[target-v2]:
return [i2, hashtable[target - v2]]
def twoSum2(self, nums, target): #56ms
hashtable = {}
for i,v in enumerate(nums):
if v in hashtable:
return[hashtable[v],i]
else:
hashtable[target - v] = i
学习心得
熟悉了字典的使用,学会了使用py中的dict和enumerate()来构建哈希表。
Leetcode (202) 快乐数
题目
编写一个算法来判断一个数是不是“快乐数”。 一个“快乐数”定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 1,也可能是无限循环但始终变不到 1。如果可以变为 1,那么这个数就是快乐数。
输入: 19
输出: true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
分析
实际上非快乐数始终为一个循环:4 → 16 → 37 → 58 → 89 → 145 → 42 → 20 → 4。
首先,初始化一个py中的集合set,通过一个循环:将该数进行快乐计算:借用map函数分割然后计算平方和,赋值给n,若所得到n不在已有的set中,则将该数加入set并再次循环。
若该数不是快乐数,则循环一轮后遇到自己,输出。
若该数是快乐数,会得到1,当遇到两次1时,输出。
最后判断输出的数是否为1。
代码
class Solution:
def isHappy(self, n): # 60ms
a = set()
while n not in a:
a.add(n)
n = sum(map(lambda x: x**2, map(int, str(n))))
return n == 1
学习心得
set(): 创建一个无序不重复元素集
a.add(x):给集合a添加元素 x
学习了map()函数的使用,包括使用lambda匿名函数进行“快乐计算”,以及给列表中的元素进行类型转换。