[알고리즘/인프런] K번째 큰 수

문제

---

현수는 1부터 100사이의 자연수가 적힌 N장의 카드를 가지고 있습니다. 같은 숫자의 카드가 여러장 있을 수 있습니다. 현수는 이 중 3장을 뽑아 각 카드에 적힌 수를 합한 값을 기록하려 고 합니다. 3장을 뽑을 수 있는 모든 경우를 기록합니다.

 

기록한 값 중 K번째로 큰 수를 출력 하는 프로그램을 작성하세요. 만약 큰 수부터 만들어진 수가 25 25 23 23 22 20 19......이고 K값이 3이라면 K번째 큰 값 은 22입니다.

 

풀이

---

#1 배열 사용

function solution(N, K, arr) {
  arr.sort((a, b) => b - a);
  let cnt = 0;
  const picked = [];

  
  for (let i = 0; i < N; i++) {
    for (let j = i + 1; j < N; j++) {
      for (let k = j + 1; k < N; k++) {
        const sum = arr[i] + arr[j] + arr[k];
        if (!picked.includes(sum)) {
          picked.push(sum);
          cnt++;
        }

        if (cnt === K) {
          return picked.at(-1);
        }
      }
    }
  }
}

let arr = [13, 15, 34, 23, 45, 65, 33, 11, 26, 42];
console.log(solution(10, 3, arr));

합이 동일한 값인 경우 cnt를 세지 않으므로 배열 대신 아래와 같이 Set을 사용할 수도 있다. 

 

#2 Set 사용

function solution(N, K, arr) {
  arr.sort((a, b) => b - a);
  let cnt = 0;
  const picked = new Set();

  for (let i = 0; i < N; i++) {
    for (let j = i + 1; j < N; j++) {
      for (let k = j + 1; k < N; k++) {
        const sum = arr[i] + arr[j] + arr[k];
        picked.add(sum);
        cnt++;

        if (cnt === K) {
          return picked.at(-1);
        }
      }
    }
  }
}

Set 대신 #1 풀이처럼 결과값을 저장하고 arr.find나 includes 메서드를 활용해 확인할 수도 있다. 하지만 이 방법은 배열 내 요소 전체를 검사해 중복 값을 찾기 때문에 Set보다 성능에서 떨어진다. Set은 값의 유일무이함을 확인하는 데 최적화되어있다고 한다. 

 

https://ko.javascript.info/map-set

맵과 셋