js 如何使用intersection获取数组交集

intersection

Set

Set

filter

Set

Set

has()

includes()

Set

has()

/**
 * 获取两个数组的交集（适用于原始值，如数字、字符串）
 * @param {Array} arr1 第一个数组
 * @param {Array} arr2 第二个数组
 * @returns {Array} 两个数组的交集
 */
const getIntersection = (arr1, arr2) => {
  // 为了性能考虑，将较短的数组转换为Set，或者干脆默认转换第二个数组
  // 这样在filter时，Set的has操作会更有效率
  const set2 = new Set(arr2); 

  // 使用filter过滤arr1中的元素，只保留set2中存在的
  const intersection = arr1.filter(item => set2.has(item));

  return intersection;
};

// 示例用法：
// const arrA = [1, 2, 3, 4, 5];
// const arrB = [3, 4, 5, 6, 7];
// console.log(getIntersection(arrA, arrB)); // 输出: [3, 4, 5]

// 另一个例子
// const arrC = ['apple', 'banana', 'orange'];
// const arrD = ['grape', 'banana', 'kiwi', 'apple'];
// console.log(getIntersection(arrC, arrD)); // 输出: ['apple', 'banana']

filter

includes

includes

Set

Set

Set

Set

set.has(element)

includes()

filter

Set

Set

filter

Set

Set

includes

{ id: 1 } === { id: 1 }

false

Set

includes

id

uuid

name

/**
 * 获取包含对象的数组交集，根据指定键进行比较
 * @param {Array<Object>} arr1 第一个对象数组
 * @param {Array<Object>} arr2 第二个对象数组
 * @param {string} key 用于比较对象的唯一键名 (e.g., 'id', 'name')
 * @returns {Array<Object>} 两个数组的交集对象
 */
const getObjectIntersection = (arr1, arr2, key) => {
  if (!key) {
    console.warn("未指定用于比较的键名，可能导致非预期结果。");
    // 也可以选择抛出错误或者默认使用一个通用比较
    return []; 
  }

  // 同样，为了效率，我们先构建一个Set，但这次Set里存储的是用于比较的键值
  const set2Keys = new Set(arr2.map(item => item[key]));

  // 过滤arr1中的对象，如果其指定键的值存在于set2Keys中，则保留
  const intersection = arr1.filter(item => set2Keys.has(item[key]));

  return intersection;
};

// 示例用法：
// const users1 = [
//   { id: 1, name: 'Alice' },
//   { id: 2, name: 'Bob' },
//   { id: 3, name: 'Charlie' }
// ];

// const users2 = [
//   { id: 2, name: 'Bob' },
//   { id: 3, name: 'Charlie' },
//   { id: 4, name: 'David' }
// ];

// console.log(getObjectIntersection(users1, users2, 'id'));
/*
输出:
[
  { id: 2, name: 'Bob' },
  { id: 3, name: 'Charlie' }
]
*/

id

key

Set

Set

Set

key