[Coding Test] 두 개 뽑아서 더하기

프로그래머스 코딩테스트 1단계 : 두 개 뽑아서 더하기

문제 설명

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

문제 풀이

• itertools 라이브러리를 이용해 조합을 사용
• 반복문으로 numbers 중 2개의 수를 뽑아 answer에 뽑은 두 수를 더해 추가
• 새롭게 생성된 answer에서 set함수를 이용해 중복 제거
• 중복을 제거한 answer를 오름차순으로 정렬해 출력

정답 코드

from itertools import combinations
def solution(numbers):
    answer = []

    for i in combinations(numbers, 2):
        answer.append(sum(i))

    answer = list(set(answer))  # 중복 제거
    answer.sort()  # 오름차순 정렬

    return answer

* 이 문제는 프로그래머스 코딩테스트 연습 1단계 문제입니다.