给定两个数组,写一个方法来计算他们的交集。 示例:数组
给定 nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].
复制代码
注意:bash
记住要记录数字及其出现次数,须要对这两个数组都遍历一遍,前后在容器记录nums1中出现的数字及其次数,在遍历查询容器中是否存在nums2中的数字,好在这一次不须要再末尾遍历输出,也就是说咱们只须要一个基础的键值对容器便可。ui
private static int[] interserct(int[] nums1, int[] nums2) {
// 用来暂存结果,保存数组的长度以及各元素的值
List<Integer> tmp = new ArrayList<>();
HashMap<Integer, Integer> map = new HashMap<>();
// 保存nums1中元素的个数
for (int i = 0; i < nums1.length; i++) {
Integer temp = map.get(nums1[i]);
map.put(nums1[i], (temp == null ? 1 : temp + 1));
}
// 遍历nums2,查询
for (int i = 0; i < nums2.length; i++) {
if (map.containsKey(nums2[i]) && map.get(nums2[i]) != 0) {
//存放该元素,且次数>0,则次数-1
tmp.add(nums2[i]);
map.put(nums2[i], map.get(nums2[i]) - 1);
}
}
int[] result = new int[tmp.size()];
int i = 0;
for (Integer in : tmp) {
result[i++] = in;
}
return result;
}
复制代码