天天看點

JS計算兩個數組的交集、差集、并集、補集(多種實作方式)

作者:前端開發婷婷

方法一:最普遍的做法

使用 ES5 文法來實作雖然會麻煩些,但相容性最好,不用考慮浏覽器 JavaScript 版本。也不用引入其他第三方庫。

1、直接使用 filter、concat 來計算

var a = [1,2,3,4,5]
var b = [2,4,6,8,10]
//交集
var intersect = a.filter(function(v){ return b.indexOf(v) > -1 })
//差集
var minus = a.filter(function(v){ return b.indexOf(v) == -1 })
//補集
var complement = a.filter(function(v){ return !(b.indexOf(v) > -1) })
 .concat(b.filter(function(v){ return !(a.indexOf(v) > -1)}))
//并集
var unionSet = a.concat(b.filter(function(v){ return !(a.indexOf(v) > -1)}));
console.log("數組a:", a);
console.log("數組b:", b);
console.log("a與b的交集:", intersect);
console.log("a與b的差集:", minus);
console.log("a與b的補集:", complement);
console.log("a與b的并集:", unionSet);
           

2、對 Array 進行擴充

(1)為友善使用,我們可以對數組功能進行擴充,增加一些常用的方法。

//數組功能擴充
//數組疊代函數
Array.prototype.each = function(fn){
 fn = fn || Function.K;
 var a = [];
 var args = Array.prototype.slice.call(arguments, 1);
 for(var i = 0; i < this.length; i++){
 var res = fn.apply(this,[this[i],i].concat(args));
 if(res != null) a.push(res);
 }
 return a;
};
//數組是否包含指定元素
Array.prototype.contains = function(suArr){
 for(var i = 0; i < this.length; i ++){
 	if(this[i] == suArr){
	 	return true;
	 }
 }
 return false;
}
//不重複元素構成的數組
Array.prototype.uniquelize = function(){
 var ra = new Array();
 for(var i = 0; i < this.length; i ++){
	 if(!ra.contains(this[i])){
	 	ra.push(this[i]);
	 }
 }
 return ra;
};
//兩個數組的交集
Array.intersect = function(a, b){
 return a.uniquelize().each(function(o){return b.contains(o) ? o : null});
};
//兩個數組的差集
Array.minus = function(a, b){
 return a.uniquelize().each(function(o){return b.contains(o) ? null : o});
};
//兩個數組的補集
Array.complement = function(a, b){
 return Array.minus(Array.union(a, b),Array.intersect(a, b));
};
//兩個數組并集
Array.union = function(a, b){
 return a.concat(b).uniquelize();
};
           

(2)使用樣例

var a = [1,2,3,4,5]
var b = [2,4,6,8,10]
console.log("數組a:", a);
console.log("數組b:", b);
console.log("a與b的交集:", Array.intersect(a, b));
console.log("a與b的差集:", Array.minus(a, b));
console.log("a與b的補集:", Array.complement(a, b));
console.log("a與b的并集:", Array.union(a, b));
           

方法二:使用 ES6 文法實作

(1)而在 ES6 中我們可以借助擴充運算符(…)以及 Set 的特性實作相關計算,代碼也會更加簡單些。

var a = [1,2,3,4,5]
var b = [2,4,6,8,10]
console.log("數組a:", a);
console.log("數組b:", b);
var sa = new Set(a);
var sb = new Set(b);
// 交集
let intersect = a.filter(x => sb.has(x));
// 差集
let minus = a.filter(x => !sb.has(x));
// 補集
let complement = [...a.filter(x => !sb.has(x)), ...b.filter(x => !sa.has(x))];
// 并集
let unionSet = Array.from(new Set([...a, ...b]));
console.log("a與b的交集:", intersect);
console.log("a與b的差集:", minus);
console.log("a與b的補集:", complement);
console.log("a與b的并集:", unionSet);
           

(2)借助includes以及箭頭函數,代碼也會更加簡單些。

var a = [1,2,3,4,5];
var b = [2,4,6,8,10];
//交集
var intersect = a.filter(v => b.includes(v));
//差集
var minus = a.filter(v => !b.includes(v));
//補集
var complement = a.filter(v => !b.includes(v)).concat(b.filter(v => !a.includes(v)))
//并集
var unionSet = a.concat(b.filter(v => !a.includes(v)));
console.log("a與b的交集:", intersect);
console.log("a與b的差集:", minus);
console.log("a與b的補集:", complement);
console.log("a與b的并集:", unionSet);
           

方法三:使用 jQuery 實作

引入 jQuery,那麼實作起來也很簡單。

var a = [1,2,3,4,5]
var b = [2,4,6,8,10]
console.log("數組a:", a);
console.log("數組b:", b);
// 交集
let intersect = $(a).filter(b).toArray();
// 差集
let minus = $(a).not(b).toArray();
// 補集
let complement = $(a).not(b).toArray().concat($(b).not(a).toArray());
// 并集
let unionSet = $.unique(a.concat(b));
console.log("a與b的交集:", intersect);
console.log("a與b的差集:", minus);
console.log("a與b的補集:", complement);
console.log("a與b的并集:", unionSet);
           

運作結果

JS計算兩個數組的交集、差集、并集、補集(多種實作方式)

繼續閱讀