logoProsperBao

FlattenDepth

2022-04-14 02

题目来源(type-challenges)

问题

递归拍平指定深度的数组

type case1 = FlattenDepth<[]> // []
type case2 = FlattenDepth<[1, 2, 3, 4]> // [1, 2, 3, 4]
type case3 = FlattenDepth<[1, [2]]> // [1, 2]
type case4 = FlattenDepth<[1, 2, [3, 4], [[[5]]]], 2> // [1, 2, 3, 4, [5]]
type case5 = FlattenDepth<[1, 2, [3, 4], [[[5]]]]> // [1, 2, 3, 4, [[5]]]
type case6 = FlattenDepth<[1, [2, [3, [4, [5]]]]], 3> // [1, 2, 3, 4, [5]]
type case7 = FlattenDepth<[1, [2, [3, [4, [5]]]]], 19260817> // [1, 2, 3, 4, 5]

解答 

type FlattenDepth<T extends any[], C extends number = 1, U extends any[] = []> = 
T extends [infer F,...infer R]
? F extends any[]
  ? U['length'] extends C
    ? [F, ...FlattenDepth<R, C, U>]
    :[...FlattenDepth<F, C, [0,...U]>,...FlattenDepth<R, C, U>]
  :[F,...FlattenDepth<R, C, U>]
: T;

拆分

  1. 首先利用 T extends [infer F,...infer R] 判断 T 是否为有内容的数组
  2. 如果不是有内容的数组，则直接返回 T
  3. 如果是有内容的数组，则利用 F extends any[] 判断第一个字元素 F 是否为数组
  4. 如果一个不是数组，那么只需要处理后面的数组就行了
  5. 如果第一个是数组，那么就需要看看累计的深度是否超过了指定的深度
  6. 如果累计的深度等于指定的深度，无需再处理当前的数组，直跳过当前元素处理下一个元素
  7. 如果没超过深度就要对两个数组进行处理拼接
  8. 针对第一个元素是数组的情况下需要递归处理，所以需要累积深度，其余的元素不需要累积深度，直接处理
cd ..