349.(查找问题)求两个数组公共元素

2018-03-22  本文已影响0人  Ching_Lee

使用set

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        // 将nums1中所有元素存到set中
        Set<Integer> set=new HashSet<Integer>();
        //用于存放结果的集合
         Set<Integer> resultset=new HashSet<Integer>(); 
        
        for(int i=0;i<nums1.length;i++)
            set.add(nums1[i]);
        //遍历nums2,将在nums1中的元素放到resultSet中
        for(int i=0;i<nums2.length;i++)
            if(set.contains(nums2[i]))
                resultset.add(nums2[i]);
        //结果数组,将set中的内容放到数组。
        int []res=new int[resultset.size()];
        Iterator it=resultset.iterator();
        int i=0;
        
        while(it.hasNext())
            res[i++]=(Integer)it.next();
            
        return res;
    }
}
上一篇下一篇

猜你喜欢

热点阅读