两个数组的交集
题目描述:给定两个数组,编写一个函数来计算它们的交集。
示例说明请见LeetCode官网。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解法一:数组遍历
- 首先将nums1中的所有数字放入Set中firstNums;
- 然后初始化一个result数组放最终的结果,并声明一个Set为secondNums用来放nums2和nums1重复的元素,然后,遍历nums2;
- 如果firstNums中存在且secondNums不存在,就将当前的数放入result中,并且放入secondNums中。
- 遍历完成后,返回result中的所有数字。
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
public class LeetCode_349 {
public static int[] intersection(int[] nums1, int[] nums2) {
Set<Integer> firstNums = new HashSet<>();
for (int i : nums1) {
firstNums.add(i);
}
int[] result = new int[nums1.length];
int index = 0;
Set<Integer> secondNums = new HashSet<>();
for (int i : nums2) {
if (firstNums.contains(i) && !secondNums.contains(i)) {
secondNums.add(i);
result[index++] = i;
}
}
return Arrays.copyOfRange(result, 0, index);
}
public static void main(String[] args) {
int[] nums1 = new int[]{1, 2, 2, 1}, nums2 = new int[]{2, 2};
for (int i : intersection(nums1, nums2)) {
System.out.print(i + " ");
}
}
}
【每日寄语】 世上最耀眼的光芒除了太阳还有你努力的模样。