题目:
给定两个数组,编写一个函数来计算它们的交集。
示例 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 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
我的思路:
第一反应:
1、用两个map来存两个数组的元素,并且对每个元素分别计数,然后将map1有,map2没有的元素从map1删除,再对比两者都有的元素,将元素出现次数最少的次数赋值给map1内的key。
2、将剩下的key-value存到arraylist里,value值为几,key就存几次。
3、将arraylist转成Integer【】,再将Integer【】转成int【】。
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
Map<Integer,Integer> map1 = new HashMap<>();
Map<Integer,Integer> map2 = new HashMap<>();
for (int i = 0; i < nums1.length; i++){
if ( !map1.containsKey(nums1[i]) ){
map1.put(nums1[i],1);
}else {
map1.put(nums1[i],map1.get(nums1[i])+1);
}
}
for (int i = 0; i < nums2.length; i++){
if ( !map2.containsKey(nums2[i]) ){
map2.put(nums2[i],1);
}else {
map2.put(nums2[i],map2.get(nums2[i])+1);
}
}
System.out.println(map1);
System.out.println(map2);
for (int i = 0; i < nums1.length; i++){
if (map2.containsKey(nums1[i])){
if (map2.get(nums1[i]) < map1.get(nums1[i])){
map1.put(nums1[i],map2.get(nums1[i]));
}
}else {
map1.remove(nums1[i]);
}
}
System.out.println(map1);
System.out.println(map2);
List list = new ArrayList();
for (Map.Entry<Integer,Integer> entry : map1.entrySet()){
int i = 0;
while(i < entry.getValue()){
list.add(entry.getKey());
i++;
}
}
System.out.println(list);
Integer[] res = new Integer[list.size()];
list.toArray(res);
int[] resint = new int[res.length];
for (int i = 0; i < res.length; i++){
resint[i] = res[i];
}
return resint;
}
}
这种方法用了好多次for循环,而且空间浪费的有点多。目前解题思路总是这样,有些复杂。
我觉得还可以用1、子字符串的方式;2、排序后的对比方式;来解决
参考他人做法:
方法一:
排序,然后直接判断nums1[i]、nums2[j]相等与否,如果相等,i++,j++,k++,k用来在nums1上覆盖新的数组数据,节省空间和改变成本。
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
int i = 0;
int j = 0;
int k = 0;
while (i < nums1.length && j < nums2.length) {
if (nums1[i] == nums2[j]) {
nums1[k] = nums1[i];
i++;
j++;
k++;
} else if (nums1[i] < nums2[j])
i++;
else
j++;
}
int[] result = new int[k];
for (int n = 0; n < k; n++)
result[n] = nums1[n];
return result;
}
}
方法二:和我的思路类似,但是比我的思路更好一些,用的空间更少,只用了一个map,加上一个限制条件就可以了,不错的想法。
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
if (nums1.length == 0 || nums2.length == 0) {
return new int[0];
}
List<Integer> listResult = new ArrayList<>();
Map<Integer, Integer> map = new HashMap<>(nums1.length);
for (int i : nums1) {
if (map.containsKey(i)) {
map.put(i, map.get(i) + 1);
continue;
}
map.put(i, 1);
}
for (int i = 0; i < nums2.length; i++) {
if (map.containsKey(nums2[i]) && map.get(nums2[i]) > 0) {
listResult.add(nums2[i]);
map.put(nums2[i], map.get(nums2[i]) - 1);
}
}
int[] result = new int[listResult.size()];
int count = 0;
for (int i :
listResult) {
result[count] = i;
count++;
}
return result;
}
}