코딩테스트

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

til_t 2021. 3. 27. 01:55

import itertools

def solution(numbers):
    answer = []
    temp = []
    temp = itertools.combinations(range(len(numbers)), 2)
    for i in temp:
        answer.append(numbers[i[0]] + numbers[i[1]])
    answer = sorted(list(set(answer)))
    return answer

itertools의 조합 함수 combinations를 사용해 인덱스를 2개 뽑고, 뽑은 인덱스를 사용해 각 값을 더한 수들을 answer 리스트에 append한 후 중복 제거를 위해 set을 이용한다.

오름차순 정렬이니 sorted를 사용해 정렬한다.