2019-04-09
描述
使用函数把集合中的元素分为两组,可通过函数来指定输入其中的元素被划分到哪一组。如果该函数返回真值,集合中的元素就属于第一组,否则就属于第二组。
提示
- 使用
Array.prototype.reduce()
和Array.prototype.push()
把元素添加到不同的组中。 - 根据集合中的每一个元素在
fn
中的返回值来进行分组。
代码
const bifurcateBy = (arr, fn) =>
arr.reduce((acc, val, i) => (acc[fn(val, i) ? 0 : 1].push(val), acc), [[], []]);
示例
根据元素在同一个函数中的运行结果来进行分组:
bifurcateBy(['beep', 'boop', 'foo', 'bar'], x => x[0] === 'b'); // [ ['beep', 'boop', 'bar'], ['foo'] ]