[js] 무작위로 K개의 수 뽑기 (lv.0, 정답률 80%)

sookyoung.k·2024년 5월 9일
0
post-thumbnail

랜덤으로 서로 다른 k개의 수를 저장한 배열을 만드려고 합니다. 적절한 방법이 떠오르지 않기 때문에 일정한 범위 내에서 무작위로 수를 뽑은 후, 지금까지 나온적이 없는 수이면 배열 맨 뒤에 추가하는 방식으로 만들기로 합니다.

이미 어떤 수가 무작위로 주어질지 알고 있다고 가정하고, 실제 만들어질 길이 k의 배열을 예상해봅시다.

정수 배열 arr가 주어집니다. 문제에서의 무작위의 수는 arr에 저장된 순서대로 주어질 예정이라고 했을 때, 완성될 배열을 return 하는 solution 함수를 완성해 주세요.

단, 완성될 배열의 길이가 k보다 작으면 나머지 값을 전부 -1로 채워서 return 합니다.

제한사항

  • 1 ≤ arr의 길이 ≤ 100,000
    - 0 ≤ arr의 원소 ≤ 100,000
  • 1 ≤ k ≤ 1,000

나의 풀이

function solution(arr, k) {
    let remove = new Set(arr);
    let result = Array.from(remove).slice(0, k);
    
    while (result.length < k) {
        result.push(-1)
    }
    return result;
}

6점 받아서 기분 개째짐 ^^

  • 중복을 제거하기 위해 Set()을 쓰고, 이를 result라는 변수에 배열로 바꿔서 저장한다.
  • slice를 통해 k 길이만큼 배열이 채워지도록 한다.
  • 하지만 result의 길이가 k보다 작으면 k 크기만큼 되도록 -1을 채워준 후 반환한다.

다른 풀이 1

function solution(arr, k) {
  const set = new Set(arr);
  return set.size < k ? [...set, ...Array(k - set.size).fill(-1)] : [...set].slice(0, k);
}

헉... 이걸 한 번에... Set을 통해 중복을 제거한 후 삼항 연산자로 바로 반환해준다.

  • set.size가 k보다 작을 경우 바로 배열에다가 [...set, ...Array(k-set.size).fill(-1)]로 배열을 만들어서 리턴하고 아니면 그냥 k길이만큼 배열 반환... 대괄호로 바로 배열을 만들 수 있군!

다른 풀이 2

const solution = (arr, k) => {
    const set = new Set(arr);
    const result = [...set].slice(0,k);
    while(result.length !== k){
        result.push(-1);
    }
    return result;
}

나랑 같은 풀이인데 그냥 while문 조건이 다를 뿐~

profile
영차영차 😎

0개의 댓글