정수 배열 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
로 오름차순으로 정렬해준다.