LeetCode哈希表篇【两个数组的交集】

力扣题目链接(opens new window)

题意：给定两个数组，编写一个函数来计算它们的交集。

说明： 输出结果中的每个元素一定是唯一的。 我们可以不考虑输出结果的顺序。

正解

一个简单的找交集问题，借用set的不重复性，统计两个数值中唯一出现的值，最后求交集。

class Solution {
       public int[] intersection(int[] nums1, int[] nums2) {
        HashSet<Integer> set1 = new HashSet();
        HashSet<Integer> set2 = new HashSet();
        HashSet<Integer> set3 = new HashSet();
        for (int i = 0; i < nums1.length; i++) {
            if (!set1.contains(nums1[i])) {
                set1.add(nums1[i]);
            }
        }
        for (int i = 0; i < nums2.length; i++) {
            if (!set2.contains(nums2[i])) {
                set2.add(nums2[i]);
            }
        }

        ArrayList <Integer>arrayList=new ArrayList(set1);//借用ArrayList获取set的值
        for (int i = 0; i < arrayList.size(); i++) {
            if (set2.contains(arrayList.get(i))){
                set3.add(arrayList.get(i));
            }
        }
        Object[] a =set3.toArray();
        int aa[]=new int[a.length];
        for (int i = 0; i < a.length; i++) {
            aa[i]=(Integer) a[i];
        }
        return aa;
    }
}

简化版

import java.util.HashSet;
import java.util.Set;

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        if (nums1 == null || nums1.length == 0 || nums2 == null || nums2.length == 0) {
            return new int[0];
        }
        Set<Integer> set1 = new HashSet<>();
        Set<Integer> resSet = new HashSet<>();
        //遍历数组1
        for (int i : nums1) {
            set1.add(i);
        }
        //遍历数组2的过程中判断哈希表中是否存在该元素
        for (int i : nums2) {
            if (set1.contains(i)) {
                resSet.add(i);
            }
        }
        int[] resArr = new int[resSet.size()];
        int index = 0;
        //将结果几何转为数组
        for (int i : resSet) {
            resArr[index++] = i;
        }
        return resArr;
    }
}