JS数组求并集，交集和差集(ES7/ES6/ES5)

需求场景

最近，自己项目中有一些数组操作，涉及到一些数学集的运算，趁着完成后总结一下。

简化问题之后，现有两数组a = [1, 2, 3]，b = [2, 4, 5]，求a，b数组的并集，交集和差集。

方法选择

JS在ES6，ES7之后，新增了一些数组方法，如果能够使用，那是极好的，毕竟自己写封装函数还是比较辛苦的。

ES7

ES7新增了一个Array.prototype.includes的数组方法，用于返回一个数组是否包含指定元素，结合filter方法。

var boolean = array.includes(searchElement[, fromIndex])

• // 并集
• let union = a.concat(b.filter(v => !a.includes(v))) // [1,2,3,4,5]
•  
• // 交集
• let intersection = a.filter(v => b.includes(v)) // [2]
•  
• // 差集
• let difference = a.concat(b).filter(v => !a.includes(v) || !b.includes(v)) // [1,3,4,5]

js

ES6

ES6中新增的一个Array.from方法，用于将类数组对象和可遍历对象转化为数组。只要类数组有length长度，基本都可以转化为数组。结合Set结构实现数学集求解。

Array.from(arrayLike[, mapFn[, thisArg]])

• let aSet = new Set(a)
• let bSet = new Set(b)
• // 并集
• let union = Array.from(new Set(a.concat(b))) // [1,2,3,4,5]
•  
• // 交集
• let intersection = Array.from(new Set(a.filter(v => bSet.has(v)))) // [2]
•  
• // 差集
• let difference = Array.from(new Set(a.concat(b).filter(v => !aSet.has(v) || !bSet.has(v)))) // [1,3,4,5]

js

ES5

ES5可以利用filter和indexOf进行数学集操作，但是，由于indexOf方法中NaN永远返回-1，所以需要进行兼容处理。

• 不考虑NaN(数组中不含NaN)

• // 并集
• var union = a.concat(b.filter(function(v) {
• return a.indexOf(v) === -1})) // [1,2,3,4,5]
•  
• // 交集
• var intersection = a.filter(function(v){ return b.indexOf(v) > -1 }) // [2]
•  
• // 差集
• var difference = a.filter(function(v){ return b.indexOf(v) === -1 }).concat(b.filter(function(v){ return a.indexOf(v) === -1 })) // [1,3,4,5]

js

• 考虑NaN

• var aHasNaN = a.some(function(v){ return isNaN(v) })
• var bHasNaN = b.some(function(v){ return isNaN(v) })
•  
• // 并集
• var union = a.concat(b.filter(function(v) {
• return a.indexOf(v) === -1 && !isNaN(v)})).concat(!aHasNaN & bHasNaN ? [NaN] : []) // [1,2,3,4,5]
•  
• // 交集
• var intersection = a.filter(function(v){ return b.indexOf(v) > -1 }).concat(aHasNaN & bHasNaN ? [NaN] : []) // [2]
•  
• // 差集
• var difference = a.filter(function(v){ return b.indexOf(v) === -1 && !isNaN(v) }).concat(b.filter(function(v){ return a.indexOf(v) === -1 && !isNaN(v) })).concat(aHasNaN ^ bHasNaN ? [NaN] : []) // [1,3,4,5]

js

结语

由于JS语言的特殊性，NaN在数组的数学集操作中有不少问题，好在ES6和ES7中，新的数组方法解决了部分情况。单从简洁性来看，ES7的方法最简洁明了。