给定两个数组,编写一个函数来计算它们的交集。
public int[] intersect(int[] nums1, int[] nums2) {
if (nums1.length > nums2.length) {
intersect(nums2, nums1);
}
List<Integer> list = new ArrayList<>();
for (int num : nums1) {
list.add(num);
}
int[] res = new int[nums1.length];
int index = 0;
for (int num : nums2) {
if (list.contains(num)) {
res[index++] = num;
//remove 是按照下标进行删除的,所以要先获取所删除元素的下标
list.remove(list.indexOf(num));
}
}
return Arrays.copyOfRange(res, 0, index);
}
public int[] intersect(int[] nums1, int[] nums2) {
//短数组变成nums1
if (nums1.length > nums2.length) {
intersect(nums2, nums1);
}
//将较短的数组存入Hash表中,有利于节省内存
Map<Integer, Integer> map = new HashMap<>();
for (int num : nums1) {
int count = map.getOrDefault(num, 0) + 1;
map.put(num, count);
}
int[] res = new int[nums1.length];
int index = 0;
for (int num : nums2) {
int count = map.getOrDefault(num, 0);
if (count > 0) {
count--;
res[index++] = num;
map.put(num, count);
}
}
return Arrays.copyOfRange(res, 0, index);
}
原文:https://www.cnblogs.com/init-qiancheng/p/14588272.html