过滤数组中的元素

说在前面

🎈不知道大家对于算法的学习是一个怎样的心态呢？为了面试还是因为兴趣？不管是出于什么原因，算法学习需要持续保持。

题目描述

给定一个整数数组 arr 和一个过滤函数 fn，并返回一个过滤后的数组 filteredArr 。

fn 函数接受一个或两个参数：

arr[i] - arr 中的数字
i - arr[i] 的索引
filteredArr 应该只包含使表达式 fn(arr[i], i) 的值为 真值 的 arr 中的元素。真值 是指 Boolean(value) 返回参数为 true 的值。

请在不使用内置的 Array.filter 方法的情况下解决该问题。

示例 1：

输入：arr = [0,10,20,30], fn = function greaterThan10(n) { return n > 10; }
输出： [20,30]
解释：
const newArray = filter(arr, fn); // [20, 30]
过滤函数过滤掉不大于 10 的值

示例 2：

输入：arr = [1,2,3], fn = function firstIndex(n, i) { return i === 0; }
输出：[1]
解释：
过滤函数 fn 也可以接受每个元素的索引
在这种情况下，过滤函数删除索引不为 0 的元素

示例 3：

输入：arr = [-2,-1,0,1,2], fn = function plusOne(n) { return n + 1 }
输出：[-2,0,1,2]
解释：
像 0 这样的假值应被过滤掉

提示：

• 0 <= arr.length <= 1000
• -10^9 <= arr[i] <= 10^9

解题思路

题目的意思其实就是要我们实现一个filter方法。

函数接受两个参数：arr是待过滤的数组，fn是一个回调函数，用于判断元素是否应该被保留。

在函数内部，首先创建了一个空数组res，用于存放符合条件的元素。

然后，通过forEach方法遍历数组arr。对于数组中的每个元素，都调用回调函数fn，并传入元素本身以及元素的索引。

如果回调函数fn返回值为true，就说明该元素符合条件，将其添加到结果数组res中。

最后，返回结果数组res，即过滤后的数组。

这段代码的时间复杂度取决于数组arr的长度，为O(n)，其中n是数组的长度。

AC代码

/**
 * @param {number[]} arr
 * @param {Function} fn
 * @return {number[]}
 */
var filter = function(arr, fn) {
    const res = [];
    arr.forEach((item,index)=>{
        if(fn(item,index)) res.push(item);
    })
    return res;
};

公众号

关注公众号『前端也能这么有趣』，获取更多有趣内容。

说在后面

🎉 这里是 JYeontu，现在是一名前端工程师，有空会刷刷算法题，平时喜欢打羽毛球 🏸 ，平时也喜欢写些东西，既为自己记录 📋，也希望可以对大家有那么一丢丢的帮助，写的不好望多多谅解 🙇，写错的地方望指出，定会认真改进 😊，偶尔也会在自己的公众号『前端也能这么有趣』发一些比较有趣的文章，有兴趣的也可以关注下。在此谢谢大家的支持，我们下文再见 🙌。