1. 程式人生 > >【JS】兩個陣列的交集 II #排序 #雜湊 #雙指標 #二分查詢

【JS】兩個陣列的交集 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 的元素儲存在磁碟上,磁碟記憶體是有限的,並且你不能一次載入所有的元素到記憶體中,你該怎麼辦?



解法一:

使用indexOf()的查詢功能,
遍歷nums1,用遍歷出的每個元素對比nums2,
當找不到相同的,就刪除nums1的元素,若找到相同的,就刪除nums2對應的元素,
返回nums1

2018/11/20 Mon

var intersect = function(nums1, nums2) {
    for(var i=0;i<nums1.length;i++){
            var pos = nums2.indexOf(nums1[i]);
            if(pos>-1) nums2.splice(pos,1)
            if(pos==-1){
                nums1.splice(i,1)
                i--
            }

    }
    return nums1;
}

61 / 61 個通過測試用例
執行用時:84 ms

or

也可以把相同的元素存入一個新的陣列

var intersect = function(nums1, nums2) {
    var pos,arr=[]
    for(var i=0;i<nums1.length;i++){
        pos = nums2.indexOf(nums1[i]);
        if(pos>-1){
            nums2.splice(pos,1);
            arr.push(nums1[i])
        } 
    }
    return arr;
}

解法二:

-如果給定的陣列已經排好序呢?你將如何優化你的演算法?
-如果 nums1 的大小比 nums2 小很多,哪種方法更優?
遍歷nums1,利用二分查詢nums2,相等的元素寫入新陣列

var intersect = function(nums1, nums2) {
    nums1=nums1.sort(sortName)
    nums2=nums2.sort(sortName)

    var arr=[]
    for(var i=0;i<nums1.length;i++){
        var lo=0,hi=nums2.length-1,mid
        while(lo<=hi){
            mid=Math.ceil((lo+hi)/2)
            if(nums2[mid]==nums1[i]) {
                arr.push(nums1[i])
                nums2.splice(mid,1)
                break;
            }else if(nums2[mid]>nums1[i]){
                hi=mid-1
            }else{
                lo=mid+1
            }
        }

    }
    return arr;
};
var sortName=function(a,b){return a-b}

61 / 61 個通過測試用例
執行用時:100 ms