349. 两个数组的交集
给定两个数组,编写一个函数来计算它们的交集。
例子
例子1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
例子2:
输入:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
代码
public int[] intersection(int[] nums1, int[] nums2) {
Map<Integer,Integer> map = new HashMap<Integer, Integer>();
LinkedList <Integer> l= new<Integer> LinkedList();
int index = 0;
for (int i = 0; i < nums1.length; i++) {
map.put(nums1[i],0);
}
for (int i = 0; i < nums2.length; i++) {
if(map.get(nums2[i]) != null && map.get(nums2[i])!=1){
l.add(nums2[i]);
map.put(nums2[i],1);
}
}
int []nums3 = new int[l.size()];
for (int i = 0; i < l.size(); i++) {
nums3[i] = l.get(i);
}
return nums3;
}
思路
(1)将a数组放在map里面
(2)用b数组去查找和map当中重复的元素
(3)用LinkedList l1存放重复元素
(4)将l1转换成数组
注意事项
待优化
可以尝试使用set