给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
C++
class Solution {
public:
vector intersect(vector& nums1, vector& nums2)
{
vector res;
map temp;
for(int i=0;i0)
{
res.push_back(nums2[i]);
temp[nums2[i]]--;
}
}
return res;
}
};
python(方法一)
class Solution:
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
temp=[]
for i in range(0,len(nums2)):
if nums2[i] in nums1:
temp.append(nums2[i])
nums1.remove(nums2[i])
i -= 1
return temp
python(方法二)
class Solution:
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
dic=dict()
res=[]
for i in range(0,len(nums1)):
if nums1[i] not in dic:
dic[nums1[i]] = 1
else:
dic[nums1[i]] += 1
for i in range(0,len(nums2)):
if nums2[i] in dic:
if(dic[nums2[i]]>0):
res.append(nums2[i])
dic[nums2[i]]-=1
return res