小知识,大挑战!本文正在参与“程序员必备小知识”创作活动
5894. 至少在两个数组中出现的值
给你三个整数数组 nums1、nums2 和 nums3 ,请你构造并返回一个 不同 数组,且由 至少 在 两个 数组中出现的所有值组成 。 数组中的元素可以按 任意 顺序排列。
示例 1:
输入: nums1 = [1,1,3,2], nums2 = [2,3], nums3 = [3]
输出: [3,2]
解释: 至少在两个数组中出现的所有值为:
- 3 ,在全部三个数组中都出现过。
- 2 ,在数组 nums1 和 nums2 中出现过。
示例 2:
输入: nums1 = [3,1], nums2 = [2,3], nums3 = [1,2]
输出: [2,3,1]
解释: 至少在两个数组中出现的所有值为:
- 2 ,在数组 nums2 和 nums3 中出现过。
- 3 ,在数组 nums1 和 nums2 中出现过。
- 1 ,在数组 nums1 和 nums3 中出现过。
示例 3:
输入: nums1 = [1,2,2], nums2 = [4,3,3], nums3 = [5]
输出: []
解释: 不存在至少在两个数组中出现的值。
提示:
1 <= nums1.length, nums2.length, nums3.length <= 1001 <= nums1[i], nums2[j], nums3[k] <= 100
解题思路
- 使用一个set1记录出现的nums1中的元素,使用set2记录出现在nums2中的元素
- 遍历nums2,检查当前遍历的元素是否出现在set1中,如果在set1存在,说明该元素满足至少在两个数组中出现过的条件,因此可以加入到结果中
- 遍历nums3,检查当前遍历的元素是否出现在set1或者set2中,如果在set1或者set2存在,说明该元素满足至少在两个数组中出现过的条件,因此可以加入到结果中
代码
class Solution {
public List<Integer> twoOutOfThree(int[] nums1, int[] nums2, int[] nums3) {
HashSet<Integer> res = new HashSet<>();
HashSet<Integer> set1 = new HashSet<>();
for (int i : nums1) {
set1.add(i);
}
HashSet<Integer> set2 = new HashSet<>();
for (int i : nums2) {
if (set1.contains(i))
res.add(i);
else set2.add(i);
}
for (int i : nums3) {
if (set1.contains(i)||set2.contains(i))
res.add(i);
}
return new ArrayList<>(res);
}
}