题目描述:
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解法:
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
HashMap<Integer,Integer> map = new HashMap<>();
int len = 0;
for (int i = 0,j = 0; i < nums1.length && j < nums2.length; ) {
if(nums1[i] == nums2[j]){
map.put(nums1[i],map.getOrDefault(nums1[i],0)+1);
len++;
i++;
j++;
}
else if(nums1[i] < nums2[j]){
i++;
}
else j++;
}
int[] res = new int[len];
int index = 0;
for (Integer e : map.keySet()) {
for (Integer i = 0; i < map.get(e); i++) {
res[index++] = e;
}
}
return res;
}
}
LinkedList解法:
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
List<Integer> list = new LinkedList<>();
for (int i = 0,j = 0; i < nums1.length && j < nums2.length; ) {
if(nums1[i] == nums2[j]){
list.add(nums1[i]);
i++;
j++;
}
else if(nums1[i] < nums2[j]){
i++;
}
else j++;
}
int[] res = new int[list.size()];
int index = 0;
for (Integer e : list) {
res[index++] = e;
}
return res;
}
}
不排序,使用LinkedList集合:
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
List<Integer> list1 = new LinkedList<>();
List<Integer> list2 = new LinkedList<>();
for (int i : nums1) {
list1.add(i);
}
for (int i : nums2) {
if(list1.contains(i)){
list2.add(i);
list1.remove(Integer.valueOf(i));
}
}
int[] res = new int[list2.size()];
int index = 0;
for (Integer e : list2) {
res[index++] = e;
}
return res;
}
}