跳转至

2634. 过滤数组中的元素

题目描述

给定一个整数数组 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
  • -109 <= arr[i] <= 109

解法

方法一:遍历

我们遍历数组 $arr$,对于每个元素 $arr[i]$,如果 $fn(arr[i], i)$ 为真,则将其加入答案数组中。最后返回答案数组即可。

时间复杂度 $O(n)$,其中 $n$ 为数组 $arr$ 的长度。忽略答案的空间消耗,空间复杂度 $O(1)$。

1
2
3
4
5
6
7
8
9
function filter(arr: number[], fn: (n: number, i: number) => any): number[] {
    const ans: number[] = [];
    for (let i = 0; i < arr.length; ++i) {
        if (fn(arr[i], i)) {
            ans.push(arr[i]);
        }
    }
    return ans;
}

评论