본문 바로가기
개발/알고리즘

[프로그래머스][LEVEL1] 두개 뽑아서 더하기

by ISA(류) 2021. 8. 18.

# 문제 원문

정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요.


제한사항

  • numbers의 길이는 2 이상 100 이하입니다.
    • numbers의 모든 수는 0 이상 100 이하입니다.

입출력 예

numbers / result

[2,1,3,4,1] [2,3,4,5,6,7]
[5,0,2,7] [2,5,7,9,12]

입출력 예 설명

입출력 예 #1

  • 2 = 1 + 1입니다. (1이 numbers에 두 개 있습니다.)
  • 3 = 2 + 1 입니다.
  • 4 = 1 + 3입니다.
  • 5 = 1 + 4 = 2 + 3입니다.
  • 6 = 2 + 4입니다.
  • 7 = 3 + 4입니다.
  • 따라서 [2,3,4,5,6,7]을 return 해야 합니다.

입출력 예 #2

  • 2 = 0 + 2입니다.
  • 5 = 5 + 0입니다.
  • 7 = 0 + 7 = 5 + 2입니다.
  • 9 = 2 + 7입니다.
  • 12 = 5 + 7입니다.
  • 따라서 [2,5,7,9,12]를 return 해야 합니다.

# 문제 풀이

입력받은 numbers []을 순회하면서 a + b를 더한 값들을 구한 후 그 값들에서 중복을 제거하고 정렬해서 반환하는 문제이다. 중복된 값 제거는 Set 자료구조를 통해서 간단히 가능하다. Set자료구조는 이 터러블이라 반환 시에는 Array.from을 이용해서 배열로 형 변환을 하는 것에 주의해야 한다. 또한 reduce와 forEach를 이용해서 FP스타일로 코딩할 수 있다.

이 경우는 FP 스타일이 오히려 가독성이나 여러 점에서 더 이득이니 FP스타일로 풀었다.

# 설루션 플로우

1. 입력받은 numbers를 순회하면서 a + b 값들을 구해준다.

2. 구해진 result를 Set를 이용해서 중복된 값들을 제거해준다.

3. 중복이 사라진 Set를 Array.from을 통해서 배열로 변경해준다.

4. 그렇게 얻어진 배열을 sort 해준다.

5. 구해진 result를 반환한다.

1. FP + Set을 활용한 풀이.

function solution(numbers) {
    return Array.from(new Set(numbers.reduce((result, current, rdx) => {
        numbers.forEach((num, idx) => {
            if (idx !== rdx) result.push(current + num);
        });
        return result;
    }, []))).sort((a, b) => a - b);
}
반응형