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

Question:

정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요.

제한사항

• numbers의 길이는 2 이상 100 이하입니다.
  • numbers의 모든 수는 0 이상 100 이하입니다.

Example:

My answer:

def solution(numbers):
    answer = []
    for i in range(len(numbers)):
        for j in range(i+1,len(numbers)):
            answer.append(numbers[i]+numbers[j])
    answer = list(set(answer))
    answer.sort()
    return answer

Other solution:

from itertools import combinations
def solution(numbers):
    answer = set()
    for i in list(combinations(numbers,2)):
        answer.add(sum(i))
    return sorted(answer)

itertools: 파이썬 기본 라이브러리.

combinations라는 내장함수를 사용하여 인자값에 따라 해당 요소로 구할 수 있는 모든 조합을 리턴

이외에도 itertools에는 permutation(list 순열을 리턴하는 함수), product(두개 이상의 리스트에서 조합을 리턴하는 함수)등이 있다. 

combinations(numbers,2) : numbers리스트 안에 2개의 요소로 구할 수 있는 모든 조합을 반환

from itertools import combinations

# list comprehension을 이용해서 한줄로 작성
def solution(numbers):
	return sorted(list(set([sum(i,j) for i,j in combinations(numbers,2)])))