给你两个整数数组 nums1 和 nums2 ,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。
来源:力扣(LeetCode)
一、个人暴力破解
空间复杂度蛮高的,,,,
public int[] intersect(int[] nums1, int[] nums2) {
List<Integer> list = new ArrayList<Integer>();
List<Integer> listTemp = new ArrayList<Integer>();
for (int i=0; i<nums1.length; i++) {
list.add(nums1[i]);
}
for(int i=0;i<nums2.length ;i++){
if(list!=null && list.size() !=0 && list.contains(nums2[i])){
listTemp.add(nums2[i]);
list.remove(Integer.valueOf(nums2[i]));
}
}
int[] res = new int[listTemp.size()];
for (int i = 0; i < listTemp.size(); i++) {
res[i] = listTemp.get(i);
}
return res;
}
二、使用hash表
key存储元素,value存储对应的元素的重复个数
然后用第二个数组遍历,存在相同的就减少count,直到count为0,说明匹配完毕
public int[] intersect(int[] nums1, int[] nums2) {
if (nums1.length > nums2.length) {
return intersect(nums2, nums1);
}
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for (int num : nums1) {
int count = map.getOrDefault(num, 0) + 1;
map.put(num, count);
}
int[] intersection = new int[nums1.length];
int index = 0;
for (int num : nums2) {
int count = map.getOrDefault(num, 0);
if (count > 0) {
intersection[index++] = num;
count--;
if (count > 0) {
map.put(num, count);
} else {
map.remove(num);
}
}
}
return Arrays.copyOfRange(intersection, 0, index);
}
这里有两个map的api
getOrDefault
获取对应key的value。可以设置默认值
Arrays.copyOfRange
将一个原始的数组original,从下标from开始复制,复制到上标to,生成一个新的数组,用于生成数组