题目
题目来源leetcode
leetcode地址:349. 两个数组的交集,难度:简单。
题目描述(摘自leetcode):
给定两个数组,编写一个函数来计算它们的交集。 示例 1: 输入:nums1 = [1,2,2,1], nums2 = [2,2] 输出:[2] 示例 2: 输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出:[9,4] 说明: 输出结果中的每个元素一定是唯一的。 我们可以不考虑输出结果的顺序。
本地调试代码:
class Solution { public int[] intersection(int[] nums1, int[] nums2) { ... } public static void main(String[] args) { int[] nums1 = new int[]{1,2,2,1}; int[] nums2 = new int[]{2,2}; int[] arr = new Solution().intersection(nums1, nums2); System.out.println(Arrays.toString(arr)); } }
题解
NO1:set去重
思路: 该题是从数组大小来看题目并没有限制,我们就不能够使用例如"有效字母异位词"方式定义数组来进行解决了,此时我们可以借助Java给我们提供的Set集合来进行解决,首先将第一组元素存储至set1中达到去重效果,接着对第二组数组进行遍历,判断每个元素是否在set1中存在,若是存在则说明该元素是两组数据的交集。
代码:
public int[] intersection(int[] nums1, int[] nums2) { if(nums1 == null || nums2 == null || nums1.length == 0 || nums1.length == 0){ return new int[]{0}; } Set<Integer> set1 = new HashSet<>(nums1.length); Set<Integer> set2 = new HashSet<>(nums2.length); //将某个数组中的元素进行去重存储到集合1中 for (int i : nums1) { set1.add(i); } //在对下一组元素进行遍历时使用set1来进行判断是否存在该元素,若是存在存储至set2中,表示为交集元素 for (int i : nums2) { if(set1.contains(i)){ set2.add(i); } } //转为整型数组 int[] distinctArr = new int[set2.size()]; int i = 0; for (Integer num : set2) { distinctArr[i++] = num; } return distinctArr; }