350. Intersection of Two Arrays II

Given two arrays, write a function to compute their intersection.

Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2, 2].

就是求两个数组的交集。
我用的是Hash Table的方法:

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersect = function(nums1, nums2) {
    var num1 = nums1.length;
    var num2 = nums2.length;
    var result = [];
    var map = {};
    if (num1===0||num2===0) {
        return result;   
    }
    else if (num1>=num2) {
        for (var i = 0; i<num2; i++) {
            if (map[nums2[i]]===undefined) {
                map[nums2[i]] = [];
            }
            if (map[nums2[i]][0]!==undefined)
                map[nums2[i]][0]++;
            else 
                map[nums2[i]][0] = 1;
        }
        for (var j = 0; j<num1; j++) {
            if (map[nums1[j]]!==undefined)
                if (map[nums1[j]][1]!==undefined)
                    map[nums1[j]][1]++;
                else 
                    map[nums1[j]][1] = 1;
        }
    } else if (num1<num2) {
        for (var m = 0; m<num1; m++) {
            if (map[nums1[m]]===undefined) {
                map[nums1[m]] = [];
            }
            if (map[nums1[m]][0]!==undefined)
                map[nums1[m]][0]++;
            else 
                map[nums1[m]][0] = 1;
        }
        for (var n = 0; n<num2; n++) {
            if (map[nums2[n]]!==undefined)
                if (map[nums2[n]][1]!==undefined)
                    map[nums2[n]][1]++;
                else 
                    map[nums2[n]][1] = 1;
        }
    }
    for ( var p in map ){
        if (map[p].length===2) {
            var count = map[p][0] > map[p][1] ? map[p][1] : map[p][0];
            for (var k = 0; k<count; k++) {
                result = result.concat(parseInt(p));
            }
        }
        
    }
    return result;
     
};

还有一种方法,对于已经排过序的数组,可以使用两个指针从头开始,两个元素相等就放到结果里,不等就把指向小的的那个指针往后移一个:

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersect = function(nums1, nums2) {
    var num1 = nums1.length;
    var num2 = nums2.length;
    var result = [];
    if (num1===0||num2===0) {
        return result;   
    }
    var a = nums1.sort(function(a,b){
        return a-b;
    });

    var b = nums2.sort(function(a,b){
        return a-b;
    });

    var i = 0;
    var j = 0;
    while (i<num1&&j<num2) {
        var item1 = a[i];
        var item2 = b[j];
        if (item1==item2) {
            result.push(item1);
            i++;
            j++;
            
        } else if (item1<item2) {
            i++;
        } else {
            j++;
        }
    }
    return result;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容