今天早上参加了中科院数字所的面试,所以发题解的时间晚了一点,正好今天的题目比较简单,也算是弥补了这个时间差。
今年数字所入营了60个人,其中3/4的成员都是来自武汉、华科、成电、天大等985,我一个211的非科班(通信工程),压力还是挺大的,不说了,进入正题:
350. 两个数组的交集 II
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
- 我们可以不考虑输出结果的顺序。
这道题比较简单,我本来想两个数组排序然后用两个指针对比的,但是考虑到排序+对比浪费的时间较多,不如两次循环通过建立和遍历哈希表解决问题了。
思路:
- 选两个数组中比较小的一个建立哈希表(节省内存考虑),统计每个数出现的次数
- 遍历另一个数组,查找是否在哈希表中,在的话则出现次数-1
class Solution:
def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
hashmap = {}
output = []
if len(nums1) > len(nums2):
return self.intersect(nums2,nums1)
for i in nums1:
if i in hashmap:
hashmap[i] += 1
else:
hashmap[i] = 1
for j in nums2:
if (j in hashmap and hashmap[j] > 0):
hashmap[j] -= 1
output.append(j)
return output
时间复杂度为O( len(nums1) + len(nums2) )
空间复杂度为O ( min( len(nums1),len(nums2) ) )
PS:
有善始者实繁,能克终者盖寡,这是我在大一刚加入科协的时候部门保研同济的老老部长,也是那一届的老主席说的话,我记到现在。
在面试川大、哈威、数字所的过程中,焦虑、急躁、不安一直充斥着我的内心,希望自己能渡过这段时间。我比较喜欢的脱口秀演员呼兰好像说过这样一句话:“你可以迷茫,可以不知所措,但是你要保持前进,不要待在原地迷茫” 【好像是这么说的】,加油!