1. 程式人生 > >查詢表類演算法//兩個陣列的交集 II

查詢表類演算法//兩個陣列的交集 II

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

示例 1:

輸入: nums1 = [1,2,2,1], nums2 = [2,2]
輸出: [2,2]

示例 2:

輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
輸出: [4,9]

說明:

  • 輸出結果中每個元素出現的次數,應與元素在兩個陣列中出現的次數一致。
  • 我們可以不考慮輸出結果的順序。

進階:

  • 如果給定的陣列已經排好序呢?你將如何優化你的演算法?
  • 如果 nums1 的大小比 nums2 小很多,哪種方法更優?
  • 如果 nums2 的元素儲存在磁碟上,磁碟記憶體是有限的,並且你不能一次載入所有的元素到記憶體中,你該怎麼辦?
class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        Map<Integer, Integer> counter = new HashMap<>(); //計數器,key為陣列中的數字,value為該數字在陣列中出現的次數
        for (int i = 0; i < nums1.length; i++) {
            int num = nums1[i];
            if (counter.containsKey(num)) {
                counter.put(num, counter.get(num) + 1);
            } else {
                counter.put(num, 1);
            }
        }
        List<Integer> tempList = new ArrayList<>();
        for (int i = 0; i < nums2.length; i++) {
            int num = nums2[i];
            if (counter.containsKey(num) && counter.get(num) > 0) {
                counter.put(num, counter.get(num) - 1); //計數器中記錄該數字的次數減1
                tempList.add(num); //將該數字新增到list中
            }
        }
        int[] result = new int[tempList.size()];
        //為滿足題目返回值型別,將list轉換為int陣列
        for (int i = 0; i < result.length; i++) {
            result[i] = tempList.get(i);
        }
        return result;
    }
}
public int[] intersect(int[] nums1, int[] nums2) {
        int cur1 = 0, cur2 = 0; // 定義指標,指向陣列開始位置
        List<Integer> list = new ArrayList<>();
        while (cur1 < nums1.length && cur2 < nums2.length) { // 迴圈結束條件:任何一個指標指向對應陣列的末端
            int num1 = nums1[cur1];
            int num2 = nums2[cur2];
            if (num1 == num2) { // 重複數字,加入結果列表中
                list.add(num1);
                cur1++;
                cur2++;
            } else if (num1 < num2) { // 將cur1指標後移一位,繼續下一次比較
                cur1++;
            } else { // 將cur2指標後移一位,繼續下一次比較
                cur2++;
            }
        }
        int[] result = new int[list.size()];
        // 為滿足題目返回值型別,將list轉換為int陣列
        for (int i = 0; i < list.size(); i++) {
            result[i] = list.get(i);
        }
        return result;
    }