JavaScript 专题(九)数组中查找指定元素
上一篇文章中,我们了解了数组扁平化的思想,并学习了 lodash 是如何处理数组扁平化的。
这次我们来讨论在数组中查找元素时所用的一些方法,并且参考lodash来实现我们自己的工具方法
文章目录
一、findIndex 和 findLastIndex
1.1 findIndex
findIndex()方法返回数组中满足提供的测试函数的第一个元素的索引。若没有找到对应元素则返回-1。
const array1 = [5, 12, 8, 130, 44];
const isLargeNumber = (element) => element > 13;
console.log(array1.findIndex(isLargeNumber));
// expected output: 3
实现
Array.prototype.newFindIndex = function(callback) {
const _arr = this;
const len = _arr.length;
for (let i = 0; i < len; i++) {
if (callback(_arr[i], i, _arr)) {
return i;
}
}
return -1;
};
const array1 = [5, 12, 8, 130, 44];
const isLargeNumber = (element) => element > 13;
console.log(array1.newFindIndex(isLargeNumber));
// 3
1.2 findLastIndex
同样当我们倒叙查找首个满足条件的方法时,可以这样写:
Array.prototype.newFindlastIndex = function(callback) {
const _arr = this;
const len = _arr.length;
for (let i = len - 1; i >= 0; i--) {
if (callback(_arr[i], i, _arr)) {
return i;
}
}
return -1;
};
const array1 = [5, 12, 8, 130, 44];
const isLargeNumber = (element) => element > 13;
console.log(array1.newFindlastIndex(isLargeNumber));
// 4
上面的代码,和正序查找很相似,仅仅改变遍历的条件。
1.3 合并 findIndex 和 findLastIndex
大家可以看到,除了循环的条件不同,两个方法几乎一模一样,参考 lodash,我们将两个方法精简一下
/**
* @private
* @param {Array} array The array to inspect.
* @param {Function} predicate The function invoked per iteration.
* @param {boolean} [fromRight] 从右向左查找
* @returns {number} 返回第一个符合条件元素的下标或-1
*/
function baseFindIndex(array, predicate, fromRight) {
const {
length } = array;
let index = fromRight ? length : -1; // 确定下标的边界
while (fromRight ? index-- : ++index < length) {
// index-- 用于倒序边界
// ++index 用于正序的边界
if (predicate(array[index], index, array)) {
return index;
}
}
return -1;
}
再来看看它的兄弟——underscore
的思路就是利用传参的不同,返回不同的函数。
function createIndexFinder(dir) {
return function(array, predicate, context) {
const {
length } = array;
var index = dir > 0 ? 0 : length - 1;
for (; index >= 0 && index < length; index += dir) {
if (predicate.call(context, array[index], index, array)) return index;
}
return -1;
};
}
var findIndex = createIndexFinder(1);
var findLastIndex = createIndexFinder(-1);
关于 findIndex
我们就告一段落了~,再来看看新的场景和实现吧!
二、sortIndex
在一个排好序的数组中找到 value
对应的位置,即保证插入数组后,依然保持有序的状态。
const arr = [1, 3, 5];
sortedIndex(arr, 0); // 0
// 不需要插入arr
那么这个又该如何实现呢?
2.1 遍历
遍历大家都能想到,虽然它不一定最优解:
function sortIndex(array, value) {
for (let i = 0; i < array.length; i++) {
if (array[i] > value) {
return i;
}
}
return array.length;
}
2.2 二分法
function sortIndex(array, value) {
let low = 0,
high = array.length;
while (low < high) {
let mid = Math.floor((low + high) / 2);
if (array[mid] < value) {
low = mid + 1;
} else {
high = mid;
}
}
return high;
}
三、indexOf 和 lastIndexOf
indexOf()
:返回在数组中可以找到一个给定元素的第一个索引,如果不存在则返回-1。从数组的前面向后查找,从 fromIndex 处开始。lastIndexOf()
:返回指定元素在数组中的最后一个的索引,如果不存在则返回-1。从数组的后面向前查找,从 fromIndex 处开始。
3.1 indexOf 的第一版实现
function indexOf(array, value) {
for (let i = 0; i < array.length; i++) {
if (array[i] === value) {
return i;
}
}
return -1;
}
emmmm…在看过 findIndex 和 lastFindIndex 的实现后,indexOf 也要整整齐齐的啊~
3.2 indexOf 和 lastIndexOf 通用第一版
通过参数来创建不同的查找方法
function createIndexOf(dir) {
return function(array, value) {
let index = dir > 0 ? 0 : arr.length - 1;
for (; index >= 0 && index < arr.length; index += dir) {
if (array[i] === value) {
return i;
}
}
return -1;
};
}
3.3 indexOf 和 lastIndexOf 通用第二版
这一次,我们允许指定查找位置,我们来看看 fromIndex 的作用:
设定开始查找的位置。如果该索引值大于或等于数组长度,意味着不会在数组里查找,返回 -1。
如果参数中提供的索引值是一个负值,则将其作为数组末尾的一个抵消,即 -1 表示从最后一个元素开始查找,-2 表示从倒数第二个元素开始查找 ,以此类推。
注意:如果参数中提供的索引值是一个负值,仍然从前向后查询数组。如果抵消后的索引值仍小于 0,则整个数组都将会被查询。其默认值为 0。
function createIndexOf(dir) {
return function(array, value, fromIndex) {
// 设定开始查找的位置。如果该索引值大于或等于数组长度,意味着不会在数组里查找,返回 -1。
let length = array == null ? 0 : array.length;
let i = 0;
if (!length) return -1;
if (fromIndex >= length) return -1;
if (typeof fromIndex === "number") {
if (dir > 0) {
// 正序
// 起始点>=0,沿用起始点,否则起始点为从后向前数fromIndex
i = fromIndex >= 0 ? fromIndex : Math.max(length + fromIndex, 0);
} else {
// 倒序
// 起始点>=0,沿用起始点,否则起始点为从后向前数fromIndex
length =
fromIndex >= 0
? Math.min(fromIndex + 1, length)
: fromIndex + length + 1;
}
}
// 起始下标
for (
fromIndex = dir > 0 ? i : length - 1;
fromIndex >= 0 && fromIndex < length;
fromIndex += dir
) {
if (array[fromIndex] === item) return fromIndex;
}
return -1;
};
}
写到这里我们在数组中查找元素就结束了,自己实现的和loadsh
或underscore
还是有很大区别的,如果大家对上面三节的代码任意有更好的实现,请一定写在留言区哦~
参考
写在最后
JavaScript 系列:
关于我
- 花名:余光(沉迷 JS,虚心学习中)
- WX:j565017805
其他沉淀
这是文章所在 GitHub 仓库的传送门,如果真的对您有所帮助,希望可以点个 star,这是对我最大的鼓励 ~
来源:oschina
链接:https://my.oschina.net/u/4356468/blog/4716748