문제 설명
정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요.
제한사항
numbers의 길이는 2 이상 100 이하입니다.
numbers의 모든 수는 0 이상 100 이하입니다.
입출력 예
| numbers | result |
|---|---|
| [2,1,3,4,1] | [2,3,4,5,6,7] |
| [5,0,2,7] | [2,5,7,9,12] |
내 코드
import itertools
def solution(numbers):
num_combination = [sum(comb) for comb in itertools.combinations(numbers,2)]
return sorted(list(set(num_combination)))풀이
문제 설명에 'numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수' 해당 문장에서 combination을 활용하여 문제를 해결할 수 있음을 암시하고 있다.
따라서 numbers 배열에서 2개의 수를 뽑는것을 itertools.combinations로 모든 순서쌍을 구하고, 그 순서쌍들의 합을 sum으로 구한다.
num_combination에는 순서쌍들의 합이 담겨져있다. (중복이 제거되지 않은 값들..)
따라서 set으로 한번 더 중복을 제거해주고 sorted로 오름차순으로 정렬해준다.
