【LeetCode】349. 两个数组的交集
349. 两个数组的交集
知识点：set; 双指针
题目描述
给定两个数组，编写一个函数来计算它们的交集。
示例
输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]
输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]
解法一：set
交集，这不就是说的重复吗？果断想到set；遍历一个数组，把不重复的存入set，再遍历另一个数组，如果set里有，那就是一个交集，防止多算，把它移除，接着遍历。
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
List<Integer> list = new ArrayList<>();
Set<Integer> set = new HashSet<>();
for(Integer i : nums1){
if(!set.contains(i)) set.add(i);
}
for(Integer i : nums2){
if(set.contains(i)){
set.remove(i);
list.add(i);
}
}
return list.stream().mapToInt(Integer::valueOf).toArray();
//要学会这个list<Integer>转为int[];此外也可以这样
//index = 0;
//int[] arr = new int[list.size()];
//for(Integer i : list){
// arr[index++] = i;
//}
//return arr;
}
}
时间复杂度:O(N+M);
解法二：排序+双指针
先将两个数组排序，这样两个数组都是从小到大了，然后使用两个指针同时遍历这两个数组，如果两个指针指的元素相等就是交集加入list，如果两个指针指的值不等，则移动较小的元素，接着比较。直到有一个指针走到了头。其实就是维持了一个从小到大的顺序。这样方便比较。
注意要比较新加入的值和已经加入的值是否相同；
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
int length1 = nums1.length, length2 = nums2.length;
List<Integer> list = new ArrayList<>();
int index = 0, index1 = 0, index2 = 0;
while (index1 < length1 && index2 < length2) {
int num1 = nums1[index1], num2 = nums2[index2];
if (num1 == num2) {
// 保证加入元素的唯一性
if (list.size() == 0 || num1 != list.get(list.size() - 1)) {
list.add(num1);
}
index1++;
index2++;
} else if (num1 < num2) {
index1++;
} else {
index2++;
}
}
int[] arr = new int[list.size()];
for(Integer i : list){
arr[index++] = i;
}
return arr;
}
}
