【题目描述】:给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
解释:[4,9] 也是可通过的
提示:
1 <= nums1.length, nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 1000
个人思路与实现
- 对两个数组进行排序
- 使用双指针分别指向两个数组
- 通过比较两个指针所对应的数组中的值来实现;
- 当左指针(指向nums1)的值大于右指针(指向nums2)时,右指针移动一位
- 同理,当小于时,左指针移动
- 当等于时,则加入到集合中
- 最后集合转为数组返回
public static int[] intersection(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
int i = 0;
int j = 0;
List<Integer> list = new ArrayList<>();
Set<Integer> set = new HashSet<>();
while (i < nums1.length && j < nums2.length) {
if (nums1[i] > nums2[j]) {
j++;
} else if (nums1[i] < nums2[j]) {
i++;
} else {
list.add(nums1[i]);
set.add(nums1[i]);
i++;
j++;
}
}
int[] array = new int[set.size()];
int m = 0;
for(Integer integer : set) {
array[m] = integer;
m++;
}
return array;
}
网上题解思路与实现
- 使用到set特性,去重
- 循环nums1,传入set集合
- 循环nums2,判断nums2的值是否存在set中,若存在,则添加进一个新的set集合中
- 最后将set集合转为数组并返回。
public static int[] intersection(int[] nums1, int[] nums2) {
Set<Integer> set = new HashSet<>();
Set<Integer> resultSet = new HashSet<>();
for(int i : nums1) {
set.add(i);
}
for (int i : nums2) {
if (set.contains(i)) {
resultSet.add(i);
}
}
return resultSet.stream().mapToInt(x-> x).toArray();
}
思路的差异
- 没想到set的去重还能这样用,对set特性理解不深刻。
- 后面集合转数组的操作还是第一次见 mapToInt 这个api,还是对stream流使用的不全面。