본문 바로가기

Programming/Programmers

[프로그래머스/Python] 두 개 뽑아서 더하기(조합)

 

코딩테스트 연습 - 두 개 뽑아서 더하기

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

programmers.co.kr

오름차순으로 출력한다는 점만 주의하자.

 

import itertools
def solution(numbers):

    answer = set()
    combi = list(itertools.combinations(numbers, 2))
    for a, b in combi:
        answer.add(a+b)
    
    return sorted(list(answer))​