1. 程式人生 > >陣列//兩個陣列的交集

陣列//兩個陣列的交集

給定兩個陣列,編寫一個函式來計算它們的交集。

示例 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;
    }
}