코딩테스트
[프로그래머스] 두 개 뽑아서 더하기 (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를 사용해 정렬한다.