给定两个集合,求两个集合的交集
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
HashSet<Integer> s1 = new HashSet<Integer>();
HashSet<Integer> s2 = new HashSet<Integer>();
for(Integer x : nums1)//遍历第一个集合,将元素添加到hashset中
s1.add(x);
for(Integer x : nums2)//同上
s2.add(x);
int i = 0;
if(s1.size() > s2.size()){//选择个数较少的那个集合
int[] a = new int[s2.size()];
for(Integer x : s2)//遍历个数较少的集合
if(s1.contains(x)) a[i++] = x;//如果个数较多的集合中包括个数较少的集合中的某个元素,那么将这个元素添加到创建的集合a中,并记录个数i
return Arrays.copyOf(a,i);//返回一个个数为i的新集合,包含a中的元素
}
else{
int[] a = new int[s1.size()];
for(Integer x : s1)
if(s2.contains(x)) a[i++] = x;
return Arrays.copyOf(a,i);
}
}
}