【LeetCode】初級算法-兩個數(shù)組的交集 II

wind3110991 發(fā)布于2019-08-23 11:43 / 2243人閱讀

摘要：題目描述兩個數(shù)組的交集給定兩個數(shù)組，編寫一個函數(shù)來計算它們的交集。示例輸入輸出示例輸入輸出說明輸出結(jié)果中每個元素出現(xiàn)的次數(shù)，應(yīng)與元素在兩個數(shù)組中出現(xiàn)的次數(shù)一致。我們可以不考慮輸出結(jié)果的順序。

題目描述

兩個數(shù)組的交集 II
給定兩個數(shù)組，編寫一個函數(shù)來計算它們的交集。

示例 1:

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

示例 2:

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

說明：

輸出結(jié)果中每個元素出現(xiàn)的次數(shù)，應(yīng)與元素在兩個數(shù)組中出現(xiàn)的次數(shù)一致。
我們可以不考慮輸出結(jié)果的順序。

進(jìn)階:

如果給定的數(shù)組已經(jīng)排好序呢？你將如何優(yōu)化你的算法？
如果 nums1 的大小比 nums2 小很多，哪種方法更優(yōu)？
如果 nums2 的元素存儲在磁盤上，磁盤內(nèi)存是有限的，并且你不能一次加載所有的元素到內(nèi)存中，你該怎么辦？

思路

排序，遍歷比較
用pos1指針遍歷nums1,pos2指針遍歷nums2
如果nums1(pos1)==nums2(pos2),則放入交集nums3中;
如果nums1(pos1)>nums2(pos2),則pos1++;
否則pos2++.

JavaScript實現(xiàn)

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersect = function(nums1, nums2) {
    nums1.sort(cmp);
    nums2.sort(cmp);
    //console.log(nums1.join(","));
    //console.log(nums2.join(","));
    let pos1 = 0;
    let pos2 = 0;
    let nums3 = [];
    while(pos1 < nums1.length && pos2 < nums2.length){
        if(nums1[pos1] == nums2[pos2]){
            nums3.push(nums1[pos1]);
            pos1++;
            pos2++;
         }else if(nums1[pos1] > nums2[pos2]){
            pos2++;
         }else{
            pos1++;
         }
    }
    return nums3;
};
function cmp(x,y){
    return x-y;
}