給定兩個數組,編寫一個函數來計算它們的交集。
示例 1:
輸入: nums1 = [1,2,2,1], nums2 = [2,2] 輸出: [2]
示例 2:
輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4] 輸出: [9,4]
思路:可以利用HashSet來存儲兩個數組的交集。
1.依次遍歷兩個數組的數字,如果兩個數組中的數字相等,則存入到Set集合中。
2.定義一個新的數組,將Set集合中的數字轉到數組中。
public static int[] intersection(int[] nums1, int[] nums2) {
Set<Integer> set=new HashSet<>();
for(int i=0;i<nums1.length;i++){
for(int j=0;j<nums2.length;j++){
if(nums1[i]==nums2[j]){
set.add(nums1[i]);
}
}
}
int[] result=new int[set.size()];
int m=0;
// 將集合轉化爲數組的第一種方法可以採用Iterator迭代器的方式
// for(Iterator<Integer>it=set.iterator();it.hasNext();){
// result[m]=it.next();
// m++;
// }
for(int k:set){
result[m]=k;
m++;
}
return result;
}