题目:
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
- 我们可以不考虑输出结果的顺序。
进阶:
- 如果给定的数组已经排好序呢?你将如何优化你的算法?
- 如果 nums1 的大小比 nums2 小很多,哪种方法更优?
- 如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
代码一:
class Solution:
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
res = []
for item in nums1:
if item in nums2:
res.append(item)
nums2.remove(item)
return res
思路二:
Python的封装性太好了,接下来我们使用常规的办法用Java代码再写一次。思路是:首先把第一个数组的值存放在一个Map中,Map的key是数组的数值,Map的value是数组中当前值出现的次数。然后遍历第二个数组,如果第二个数组当前的数值在Map中且次数不小于1的话,把当前数加入到结果集中。
代码二:
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
List<Integer> res = new ArrayList<Integer>();
HashMap<Integer, Integer> hashMap = new HashMap<Integer,Integer>();
for(int i = 0;i< nums1.length;i++){
if (hashMap.containsKey(nums1[i])){
Integer value = hashMap.get(nums1[i]);
hashMap.put(nums1[i], value+1);
}
else{
hashMap.put(nums1[i], 1);
}
}
for(int j = 0;j<nums2.length;j++){
if(hashMap.containsKey(nums2[j])&&hashMap.get(nums2[j])> 0){
res.add(nums2[j]);
hashMap.put(nums2[j], hashMap.get(nums2[j])-1);
}
}
int[] result = new int[res.size()];
int i = 0;
for (Integer temp : res) {
result[i++] = temp;
}
return result;
}
}