Description
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.
返回两个列表的交集,有重复元素时,只返回一个。
解:
- 使用
hashset
,遍历其中一个数组,同时将该数组元素存入set1中。 - 之后,遍历另外一个数组,同时判断该数组遍历到的元素是否已经存在于set1中,若存在,则将该元素存入set2。
- 最后,将set2的元素存入结果数组。
java代码:
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
HashSet<Integer> set_1 = new HashSet<Integer>();
HashSet<Integer> set_2 = new HashSet<Integer>();
for (int i = 0; i < nums1.length; i++) {
set_1.add(nums1[i]);
}
for (int i = 0; i < nums2.length; i++) {
if (set_1.contains(nums2[i])){
set_2.add(nums2[i]);
}
}
int[] res = new int[set_2.size()];
int i = 0;
for (int x : set_2){
res[i] = x;
i++;
}
return res;
}
}