leetcode——349

给定两个集合,求两个集合的交集

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        HashSet s1 = new HashSet();
        HashSet s2 = new HashSet();
        for(Integer x : nums1)//遍历第一个集合,将元素添加到hashset中
            s1.add(x);
        for(Integer x : nums2)//同上
            s2.add(x);
        int i = 0;
      
        if(s1.size() > s2.size()){//选择个数较少的那个集合
            int[] a = new int[s2.size()];
            for(Integer x : s2)//遍历个数较少的集合
                if(s1.contains(x)) a[i++] = x;//如果个数较多的集合中包括个数较少的集合中的某个元素,那么将这个元素添加到创建的集合a中,并记录个数i
            return Arrays.copyOf(a,i);//返回一个个数为i的新集合,包含a中的元素
        }
        else{
            int[] a = new int[s1.size()];
            for(Integer x : s1)
                if(s2.contains(x)) a[i++] = x;
            return Arrays.copyOf(a,i);
        }
    }
}

你可能感兴趣的:(java)