leetcode (Intersection of Two Arrays)
阿新 • • 發佈:2019-01-06
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; }