每天一个lodash方法(9)

2018-08-30  本文已影响39人  公子七

Array method 系列之九 —— pull

以上四种方法非常相似,其核心都是basePullAll,以下是该方法的源码。

/**
 * @private
 * @param {Array} array The array to modify.  要修改的数组
 * @param {Array} values The values to remove. 需移除的元素
 * @param {Function} [iteratee] The iteratee invoked per element. 迭代器方法
 * @param {Function} [comparator] The comparator invoked per element. 比较器方法
 * @returns {Array} Returns `array`. 返回修改后的数组
 */
function basePullAll(array, values, iteratee, comparator) {
  const indexOf = comparator ? baseIndexOfWith : baseIndexOf
  const length = values.length

  let index = -1
  let seen = array

  if (array === values) {
    // 不是很清楚为什么要对values数组进行一波复制操作???
    values = copyArray(values)
  }
  // 如果存在迭代器方法,先对数组元素遍历处理
  if (iteratee) {
    seen = map(array, (value) => iteratee(value))
  }
  while (++index < length) {
    let fromIndex = 0
    const value = values[index]
    const computed = iteratee ? iteratee(value) : value
   // 判断需删除的元素是否在数组中,如果存在,删除;
   // fromIndex是每次查找的起始位置,初始默认为0,每次都从上次删除的位置进行查找,相比逐个遍历,降低了时间复杂度。
    while ((fromIndex = indexOf(seen, computed, fromIndex, comparator)) > -1) {
      if (seen !== array) {
        seen.splice(fromIndex, 1)
      }
      array.splice(fromIndex, 1)
    }
  }
  return array
}
上一篇下一篇

猜你喜欢

热点阅读