349_LeetCode_349 Intersection of Two Arrays 题解

Description

---

Given two arrays, write a function to compute their intersection

---

Example:

---

Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].

---

Note:

---

• Each element in the result must be unique.
• The result can be in any order.

返回两个列表的交集，有重复元素时，只返回一个。

---

解：

---

• 使用hashset，遍历其中一个数组，同时将该数组元素存入set1中。
• 之后，遍历另外一个数组，同时判断该数组遍历到的元素是否已经存在于set1中，若存在，则将该元素存入set2。
• 最后，将set2的元素存入结果数组。

---

java代码：

---

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        HashSet set_1 = new HashSet();
        HashSet set_2 = new HashSet();
        for (int i = 0; i < nums1.length; i++) {
            set_1.add(nums1[i]);
        }
        for (int i = 0; i < nums2.length; i++) {
            if (set_1.contains(nums2[i])){
                set_2.add(nums2[i]);
            }
        }
        int[] res = new int[set_2.size()];
        int i = 0;
        for (int x : set_2){
            res[i] = x;
            i++;
        }
        return res;
    }
}