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