LintCode547:两组数的交

返回两个数组的交
样例
nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2].

代码如下

public int[] intersection(int[] nums1, int[] nums2) {
        // Write your code here
        HashSet<Integer> set1 = new HashSet<Integer>();
        for(int i:nums1){
            set1.add(i);
        }
        HashSet<Integer> set2  = new HashSet<Integer>();
        for(int i:nums2){
            if(set1.contains(i)){
                set2.add(i);
            }
        }
        Iterator<Integer> iterator =  set2.iterator();
        int[] nums3 = new int[set2.size()];
        int i= 0;
        while(iterator.hasNext()){
            nums3[i++] = iterator.next();
        }
        return nums3;
    }

这里用HashSet是因为HashSet不能储存重复对象刚好满足题目要求。