본문 바로가기

알고리즘 & 코딩테스트/코딩테스트

Programmers Lv.1 "두 개 뽑아서 더하기"

Programmers Lv.1 두 개 뽑아서 더하기, 월간 코드 챌린지 시즌 1, 정답률 68%

주어진 정수배열의 요소 중 2개를 뽑아 덧셈을 한 결과를 반환하면 되는데 모든 경우를 다 체크하면 된다. 
알고있다면 조금 쉬울 수 있는데, 예전에 배웠던 '조합'을 이용하면 쉽게쉽게 구할 수 있다.

from itertools import combinations

def solution(numbers):
    return sorted(list(set([sum([i,j]) for i,j in combinations(numbers,2)])))