在javascript数组中执行Set计算

时间:2014-12-23 15:20:07

标签: javascript arrays union intersection set-theory

我有2个数组可以说:

A = [1,2,3,4,5] and B = [1,2,3,6,7]

我想执行以下'设置计算':

C = (A ∩ B)
D = A - (A ∩ B)
E = B - (A ∩ B)

本质:

C = [1,2,3]
D = [4,5]
E = [6,7]

有没有一种聪明的方法可以做到这一点,或者我将不得不用循环和ifs交叉检查每个数组成员?我不能使用外部库(如math.js或w / e)。

提前致谢。

2 个答案:

答案 0 :(得分:5)

filter()至少可以隐藏循环:

A = [1,2,3,4,5];
B = [1,2,3,6,7];

C = intersection(A, B);
D = arrayDiff(A, C);
E = arrayDiff(B, C);

console.log(JSON.stringify(C));
console.log(JSON.stringify(D));
console.log(JSON.stringify(E));

function intersection(a, b) {
  return a.filter( 
    function(el) {
      return b.indexOf(el) >= 0;
    }
  );
}

function arrayDiff(a, b) {
  return a.filter( 
    function(el) {
      return b.indexOf(el) < 0;
    }
  );
}

答案 1 :(得分:3)

从ES6开始,Javascript有一个inbuilt set object,它提供了完成上述操作的简洁方法。

&#13;
&#13;
var intersection = function(setA, setB){
  return new Set([x for (x of setA) if (setB.has(x))]);
}
var difference = function(setA, setB){
  return new Set([x for (x of setA) if (!setB.has(x))]);
}
A = new Set([1,2,3,4,5]);
B = new Set([1,2,3,6,7]);

// A ∩ B = ([1,2,3])
intersection(A, B);

// A \ B = ([4,5])
difference(A, B);

// B \ A = ([6,7])
difference(B, A);
&#13;
&#13;
&#13;