FE/type-challenge

5310. Join

최토피 2023. 8. 18. 12:46
728x90

문제

Implement the type version of Array.join, Join<T, U> takes an Array T, string or number U and returns the Array T with U stitching up.

type Res = Join<["a", "p", "p", "l", "e"], "-">; // expected to be 'a-p-p-l-e'
type Res1 = Join<["Hello", "World"], " ">; // expected to be 'Hello World'
type Res2 = Join<["2", "2", "2"], 1>; // expected to be '21212'
type Res3 = Join<["o"], "u">; // expected to be 'o'

cases

type cases = [
  Expect<Equal<Join<['a', 'p', 'p', 'l', 'e'], '-'>, 'a-p-p-l-e'>>,
  Expect<Equal<Join<['Hello', 'World'], ' '>, 'Hello World'>>,
  Expect<Equal<Join<['2', '2', '2'], 1>, '21212'>>,
  Expect<Equal<Join<['o'], 'u'>, 'o'>>,
]

문제 링크

정답

type Join<T extends unknown[], U extends string | number> = T extends [infer T1 extends string, ...infer Rest]
  ? Rest['length'] extends 0
    ? T1
    : `${T1}${U}${Join<Rest, U>}`
  : ''

풀이

조건

  1. 배열 T 요소 간 U를 추가한다.

해설

배열 T의 요소 뒤에 U 를 추가한다.

type Join<T extends unknown[], U extends string | number> = T extends [infer T1, ...infer Rest]
  ? `${T1}${U}${Join<Rest, U>}`
  : ''

T1의 타입을 string으로 추정할 수 있도록 한다.

type Join<T extends unknown[], U extends string | number> = T extends [infer T1 extends string, ...infer Rest]
  ? `${T1}${U}${Join<Rest, U>}`
  : ''

재귀로 탐색하다가 T의 길이가 1인 경우, 즉 마지막 요소의 경우를 처리해준다.

type Join<T extends unknown[], U extends string | number> = T extends [infer T1 extends string, ...infer Rest]
  ? Rest['length'] extends 0
    ? T1
    : `${T1}${U}${Join<Rest, U>}`
  : ''

혹시 오류나 개선점이 있다면 댓글 부탁드립니다.

728x90