GitHub地址
题目
给定两个数组,写一个方法来计算它们的交集。
示例
给定 nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].
注意
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
跟进
如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果nums2的元素存储在磁盘上,内存是有限的,你不能一次加载所有的元素到内存中,你该怎么办?
代码
这一题相对来说比较简单,我一开始的思路是将数组转换成list这样就可以使用list提供的contains方法了,也就是说时间复杂度为O(n),但是由于int类型的数组,没有办法直接转换成list,所以只能做了下面的转换操作,导致时间复杂度为O(3n).下面的代码LeetCode为40%。
public static int[] intersect(int[] nums1, int[] nums2) {
List list1 = new ArrayList();
List list = new ArrayList();
for (int i : nums1) {
Integer integer = i;
list1.add(integer);
}
Integer temp;
for (int i = 0; i < nums2.length; i++) {
if (list1.contains(nums2[i])) {
temp = nums2[i];
list.add(temp);
list1.remove(temp);
if (list1.size() == 0) {
break;
}
}
}
int[] nums = new int[list.size()];
for (int i = 0; i < nums.length; i++) {
nums[i] = (int) list.get(i);
}
return nums;
}
看了跟进之后想了一下,之前好像写过类似的题目,排序之后将两个数组同时遍历,比较同索引位置的元素是否相等,相等则记录下来,依次遍历两个数据中的元素,一提交居然击败了100%的LeetCode小伙伴,有点开心呀。
public static int[] intersect2(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
int[] nums = new int[nums1.length>nums2.length?nums2.length:nums1.length];
int i = 0, j = 0,k = 0;
while (i < nums1.length && j < nums2.length) {
if (nums1[i] == nums2[j]) {
nums[k++] = nums1[i];
i++;
j++;
} else if (nums1[i] > nums2[j]) {
j++;
} else {
i++;
}
}
int[] numss = new int[k];
System.arraycopy(nums,0,numss,0,k);
return numss;
}