547.两数组的交集
描述
给出两个数组,写出一个方法求出它们的交集
样例
例1:
输入: nums1 = [1, 2, 2, 1], nums2 = [2, 2],
输出: [2].
例2:
输入: nums1 = [1, 2], nums2 = [2],
输出: [2].
挑战
可以用三种不同的方法实现吗?
注意事项
- 结果中的每个元素必须是唯一的。
- 结果需要为升序。
代码部分
public class Solution {
/**
* @param nums1: an integer array
* @param nums2: an integer array
* @return: an integer array
*/
public int[] intersection(int[] nums1, int[] nums2) {
// write your code here
HashSet<Integer> set1=new HashSet<Integer>();
for(int i:nums1){
set1.add(i);
}
HashSet<Integer> set2=new HashSet<Integer>();
for(int i:nums2){
if(set1.contains(i)){
set2.add(i);
}
}
int[] result=new int[set2.size()];
int i=0;
for(int n:set2){
result[i++]=n;
}
return result;
}
}
548.两数组的交集
给定两个数组,计算两个数组的交集
样例
Example1
Input:
nums1 = [1, 2, 2, 1], nums2 = [2, 2]
Output:
[2, 2]
Example2
Input:
nums1 = [1, 1, 2], nums2 = [1]
Output:
[1]
挑战
-如果给定的数组已经排序了怎么办?如何优化算法?
-如果nums1的大小比num2的小怎么办?哪种算法更好?
-如果nums2的元素存储在磁盘上,并且内存受到限制,以至于不能一次将所有元素加载到内存中,该怎么办?
注意事项
每个元素出现次数得和在数组里一样
答案可以以任意顺序给出
public class Solution {
/**
* @param nums1: an integer array
* @param nums2: an integer array
* @return: an integer array
*/
public int[] intersection(int[] nums1, int[] nums2) {
// write your code here
List<Integer> tmp = new ArrayList<>();
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for (int i = 0; i < nums1.length; i++) {
Integer value = map.get(nums1[i]);
map.put(nums1[i], (value == null ? 0 : value) + 1);
}
for (int i = 0; i < nums2.length; i++) {
if (map.containsKey(nums2[i]) && map.get(nums2[i]) != 0) {
tmp.add(nums2[i]);
map.put(nums2[i], map.get(nums2[i]) - 1);
}
}
int[] result = new int[tmp.size()];
int k = 0;
for (Integer e : tmp)
result[k++] = e;
return result;
}
}
补充说明
集合框架的使用