Skip to content

Without

约 381 字大约 1 分钟

2022-12-01

题目

Github: Without

实现一个像 Lodash.without 函数一样的泛型 Without<T, U>,它接收数组类型的 T 和数字或数组类型的 U 为参数, 会返回一个去除 U 中元素的数组 T

type Res = Without<[1, 2], 1> // expected to be [2]
type Res1 = Without<[1, 2, 4, 1, 5], [1, 2]> // expected to be [4, 5]
type Res2 = Without<[2, 3, 2, 3, 2, 3, 2, 3], [2, 3]> // expected to be []

解题思路

此挑战很容易想到使用 infer 和递归 来实现,从 数组 T 中取出一个个元素,判断是否为 U 的元素。

但还需要注意一点,即 U 可能传入的是 数组,也可能是 非数组。因此,需要将 U 转换为 联合类型。

答案

type ToUnion<U> = U extends readonly any[] ? U[number] : U
type Without<T extends any[], U> = T extends [infer F, ...infer O]
  ? F extends ToUnion<U>
    ? Without<O, U>
    : [F, ...Without<O, U>]
  : T

验证

type 
cases
= [
Expect
<
Equal
<
Without
<[1, 2], 1>, [2]>>,
Expect
<
Equal
<
Without
<[1, 2, 4, 1, 5], [1, 2]>, [4, 5]>>,
Expect
<
Equal
<
Without
<[2, 3, 2, 3, 2, 3, 2, 3], [2, 3]>, []>>,
]

参考