定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
进阶:
如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
方法一:将第一个数组转换成hashmap, 然后对第二个数组进行遍历
public int[] intersect(int[] nums1, int[] nums2) {
Map<Integer,Integer> map = new HashMap<>();
ArrayList<Integer> arrayList = new ArrayList();
if(nums1 != null && nums1.length > 0 ){
int length = nums1.length;
for(int i = 0; i < length; i++){
if(map.get(nums1[i]) == null){
map.put(nums1[i],1);
}else{
int temp = map.get(nums1[i]) + 1;
map.replace(nums1[i],temp);
}
}
if(nums2 != null && nums2.length > 0){
int length1 = nums2.length;
for(int i = 0; i < length1; i++){
if(map.containsKey(nums2[i])){
Integer temp = map.get(nums2[i]);
if(temp > 1){
map.replace(nums2[i],temp-1);
}else{
map.remove(nums2[i]);
}
arrayList.add(nums2[i]);
}
}
}else{
return new int[]{};
}
int length2 = arrayList.size();
int[] result = new int[length2];
for(int i = 0; i < length2; i++){
result[i] = arrayList.get(i);
}
return result;
}else{
return new int[]{};
}
}
方法二:转成两个map
public int[] intersect(int[] nums1, int[] nums2) {
HashMap<Integer, Integer> x = new HashMap<>();
HashMap<Integer, Integer> y = new HashMap<>();
for (Integer num : nums1) {
if (x.containsKey(num))
x.replace(num, x.get(num) + 1);
else
x.put(num, 1);
}
int i = 0;
for (Integer num : nums2) {
if (x.containsKey(num) && x.get(num) > 0) {
y.put(i, num);
i++;
x.replace(num, x.get(num) - 1);
}
}
int[] res = new int[y.size()];
for (Integer key : y.keySet()) {
res[key] = y.get(key).intValue();
}
return res;
}
方法三:对数组进行排序,针对有序数组进行比较
public int[] intersect(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
List<Integer> list = new ArrayList<>();
for (int i = 0, j = 0; i < nums1.length && j < nums2.length; ) {
if (nums1[i] < nums2[j]) {
i++;
} else if (nums1[i] > nums2[j]) {
j++;
} else {
list.add(nums1[i]);
i++;
j++;
}
}
int[] res = new int[list.size()];
for (int i = 0; i < list.size(); i++) {
res[i] = list.get(i);
}
return res;
}