1. 程式人生 > >leetcode (Intersection of Two Arrays)

leetcode (Intersection of Two Arrays)

Title: Intersection of Two Arrays     349

Difficulty:Easy

原題leetcode地址:   https://leetcode.com/problems/intersection-of-two-arrays/

 

1.  見程式碼註釋

時間複雜度:O(n),三次一層for迴圈。

空間複雜度:O(n),申請集合set和連結串列list以及需要返回的陣列。

    /**
     * nums1放到一個set中,隊重複元素進行過濾
     * 將nums1和nums2中相同的數放到一個list中
     * 將上述list轉換成陣列返回
     * @param nums1
     * @param nums2
     * @return
     */
    public static int[] intersection(int[] nums1, int[] nums2) {

        if (nums1.length == 0 || nums2.length == 0) {
            return new int[]{};
        }

        Set<Integer> set = new HashSet<>();
        List<Integer> list = new ArrayList<>();

        for (int i = 0; i < nums1.length; i++) {
            if (set.contains(nums1[i])) {
                continue;
            }
            else {
                set.add(nums1[i]);
            }
        }

        for (int i = 0; i < nums2.length; i++) {
            if (set.contains(nums2[i])) {
                if (!list.contains(nums2[i])) {
                    list.add(nums2[i]);
                }
            }
        }

        int res[] = new int[list.size()];
        for (int i = 0; i < res.length; i++) {
            res[i] = list.get(i);
        }

        return res;

    }