题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
349、给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
说明:
- 输出结果中的每个元素一定是唯一的。
- 我们可以不考虑输出结果的顺序。
解题: 求两个数组的交集,直接先对数组排序,然后进行双指针同时遍历,把相等的元素 加入集合 中,最后转成数组返回。注意在双指针 遍历的时候,会有相同元素,使用的集合为HashSet,可以保证唯一性。
代码:
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
//排序
Arrays.sort(nums1);
Arrays.sort(nums2);
int len1 = nums1.length;
int len2 = nums2.length;
//创建HashSet
Set<Integer> set = new HashSet<Integer>();
int p = 0, q = 0;
while (p < len1 && q < len2) {
//如果两个数组当前索引下的元素相等,就加入到集合中
if (nums1[p] == nums2[q]) {
set.add(nums1[p]);
p++;
q++;
} else if (nums1[p] > nums2[q]) {//如果其中一个元素较大,那么另一个数组的索引应该增加
q++;
} else if (nums1[p] < nums2[q]) {
p++;
}
}
//创建返回数组
int[] rs = new int[set.size()];
int index = 0;
//遍历集合填充数组
for (int s: set) {
rs[index++] = s;
}
return rs;
}
}
350、给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现次数的最小值一致。
- 我们可以不考虑输出结果的顺序。
解题: 此题和上题差不多,只不过上一题要求除去重复相同的元素,而这题要求保留。思路想法和上题一直,采用排序加双指针的方式就可以了,不过此次不需要用HashSet来去重。
代码;
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
int len1 = nums1.length;
int len2 = nums2.length;
//排序
Arrays.sort(nums1);
Arrays.sort(nums2);
//节省空间,取小的来创建数组
int[] rs = new int[Math.min(len1, len2)];
int p = 0, q = 0;
int index = 0;
while (p < len1 && q < len2) {
if (nums1[p] > nums2[q]) {
q++;
} else if (nums1[p] < nums2[q]) {
p++;
} else {
rs[index++] = nums1[p];
p++;
q++;
}
}
return Arrays.copyOfRange(rs, 0, index);
}
}