陣列//兩個陣列的交集
阿新 • • 發佈:2018-11-11
給定兩個陣列,編寫一個函式來計算它們的交集。
示例 1:
輸入: nums1 = [1,2,2,1], nums2 = [2,2] 輸出: [2]
示例 2:
輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4] 輸出: [9,4]
說明:
- 輸出結果中的每個元素一定是唯一的。
- 我們可以不考慮輸出結果的順序。
class Solution { public int[] intersection(int[] nums1, int[] nums2) { List<Integer> tmp = new ArrayList<>(); Map<Integer, Integer> map = new HashMap<Integer, Integer>(); for (int i = 0; i < nums1.length; i++) { Integer value = map.get(nums1[i]); map.put(nums1[i], (value == null ? 0 : value) + 1); } for (int i = 0; i < nums2.length; i++) { if (map.containsKey(nums2[i]) && map.get(nums2[i]) != 0) { tmp.add(nums2[i]); map.put(nums2[i], map.get(nums2[i])-1); } } int[] result = new int[tmp.size()]; int i = 0; for (Integer e : tmp) result[i++] = e; return result; } }
class Solution { public int[] intersection(int[] nums1, int[] nums2) { Arrays.sort(nums1); Arrays.sort(nums2); List<Integer> tmp = new ArrayList<>(); int i = 0; int j = 0; while(i < nums1.length && j < nums2.length){ if(nums2[j] > nums1[i]){ i++; }else if(nums2[j] < nums1[i]){ j++; }else{ tmp.add(nums1[i]); i++; j++; } } int []result = new int[tmp.size()]; for(int k = 0; k < result.length; k++){ result[k] = tmp.get(k); } return result; } }