JS实现集合的交集、补集、差集、去重运算示例【ES5与ES6写法】

2023年5月28日上午8:54 • JavaScript

下面就给大家讲解一下“JS实现集合的交集、补集、差集、去重运算示例【ES5与ES6写法】”的完整攻略。

1. 集合的定义和常用运算

1.1 集合的定义

集合（Set）是一种专门存储不重复值的数据结构。它可以存储各种类型的原始值和对象引用。与数组不同的是，集合没有顺序概念，因此不能通过索引访问元素。集合中的数据也没有重复，每个元素的位置都是唯一的。

1.2 集合的常用运算

集合常用的运算有交集、补集、差集、去重四种。

交集：将两个集合的共同元素组成的集合称为它们的交集。
补集：A的补集表示在全集中不属于A的元素所组成的集合。
差集：两个集合A和B的差集，指的是由所有属于A但不属于B的元素所组成的集合。
去重：去重是指从一个集合中去掉其中重复的元素。

2. ES5实现集合的运算

2.1 创建集合

ES5中创建集合通常使用数组表示，为了避免数组中的重复元素，我们需要在创建时进行去重。

function createSet(arr) {
  var setArr = [];
  for (var i = 0; i < arr.length; i++) {
    if (setArr.indexOf(arr[i]) === -1) {
      setArr.push(arr[i]);
    }
  }
  return setArr;
}

2.2 求集合交集

求两个集合的交集，需要遍历第一个集合，同时判断它在第二个集合中是否存在，如果存在就添加到结果集中。

function intersection(setA, setB) {
  var intersectionSet = [];
  for (var i = 0; i < setA.length; i++) {
    var item = setA[i];
    if (setB.indexOf(item) !== -1) {
      intersectionSet.push(item);
    }
  }
  return intersectionSet;
}

2.3 求集合补集

求一个集合的补集，需要遍历全集，同时找出不属于它的元素。

function complement(setA, fullSet) {
  var complementSet = [];
  for (var i = 0; i < fullSet.length; i++) {
    var item = fullSet[i];
    if (setA.indexOf(item) === -1) {
      complementSet.push(item);
    }
  }
  return complementSet;
}

2.4 求集合差集

求两个集合的差集，可以先求出两个集合的交集，然后在第一个集合中排除交集中的元素。

function difference(setA, setB) {
  var intersectionSet = intersection(setA, setB);
  var differenceSet = [];
  for (var i = 0; i < setA.length; i++) {
    var item = setA[i];
    if (intersectionSet.indexOf(item) === -1) {
      differenceSet.push(item);
    }
  }
  return differenceSet;
}

2.5 去重

去重可以使用ES5的数组去重方法，将一个数组转换成集合即可。

function unique(arr) {
  return Array.from(new Set(arr));
}

3. ES6实现集合的运算

3.1 创建集合

ES6中提供了Set对象，可以方便地进行集合操作。

function createSet(arr) {
  return new Set(arr);
}

3.2 求集合交集

Set对象提供了交集运算符&，使用交集运算符即可得到两个集合的交集。

function intersection(setA, setB) {
  return new Set([...setA].filter(item => setB.has(item)));
}

3.3 求集合补集

可以通过全集减去原集合得到补集。

function complement(setA, fullSet) {
  return new Set([...fullSet].filter(item => !setA.has(item)));
}

3.4 求集合差集

也是通过交集求解。

function difference(setA, setB) {
  return new Set([...setA].filter(item => !setB.has(item)));
}

3.5 去重

使用Set对象即可去重。

function unique(arr) {
  return Array.from(new Set(arr));
}

4. 示例说明

4.1 求两个数组的交集

以下是使用ES6实现的求取两个数组的交集的代码示例。

var arr1 = [1, 2, 3, 4, 5];
var arr2 = [3, 4, 5, 6, 7];
var setA = createSet(arr1);
var setB = createSet(arr2);
var intersectionSet = intersection(setA, setB);
console.log(intersectionSet); // Set {3, 4, 5}

4.2 求两个数组的差集

以下是使用ES5实现的求取两个数组的差集的代码示例。

var arr1 = [1, 2, 3, 4, 5];
var arr2 = [3, 4, 5, 6, 7];
var setA = createSet(arr1);
var setB = createSet(arr2);
var differenceSet = difference(setA, setB);
console.log(differenceSet); // [1, 2]