LeetCode349 InsectionOfTwoArrays

LeetCode349 InsectionOfTwoArrays

题目描述

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

示例1：

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]

示例2：

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]

输出: [9,4]

说明:

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

解答

解法一：Set的使用

用两个set，一个set用来存储nums1的数组，另一个用来保存交集，然后将交集转化成数组返回。

public static int[] solve(int[] nums1, int[] nums2){
    Set<Integer> temp = new HashSet<>();
    Set<Integer> result = new HashSet<>();

    for (int i=0;i<nums1.length;i++){
        temp.add(nums1[i]);
    }
    for (int i=0; i<nums2.length;i++){
        if (temp.contains(nums2[i]))
            result.add(nums2[i]);
    }

    int[] insection = new int[result.size()];
    int index =0;
    for (int a: result) {
        insection[index++] = a;
    }

    //        Iterator<Integer> it = result.iterator();
    //        int i=0;
    //        while (it.hasNext()){
    //            insection[i] = it.next();
    //            i++;
    //        }

    return insection;
}