Given two arrays, write a function to compute their intersection.
Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].
Note:
* Each element in the result must be unique.
* The result can be in any order.
给定两个数组,计算它们的重复部分。
注意:返回结果中的元素不要重复,结果可以任意顺序组织。
思路:
利用关联容器set保存nums1的元素,对于nums2中的每个元素,检查是否在set中。
class Solution {
public:
vector intersection(vector& nums1, vector& nums2) {
unordered_set m(nums1.begin(), nums1.end());
vector res;
for (auto a : nums2)
if (m.count(a)) { //元素重复
res.push_back(a);
m.erase(a); //已经计入的不再重复计算
}
return res;
}
};
public class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
Set set1 = new HashSet<>();
Set set2 = new HashSet<>();
for(int i=0;i