일기 대신 코드 슬쩍

[프로그래머스] Lv1.두 개 뽑아서 더하기 본문

코딩테스트/프로그래머스(Python)

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

코코자 2023. 1. 29. 23:13

문제 설명

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

제한사항

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

입출력 예

numbersresult
[2,1,3,4,1][2,3,4,5,6,7]
[5,0,2,7][2,5,7,9,12]

풀이

from itertools import combinations # itertools 조합 사용
def solution(numbers):
    answer = []
    for i in combinations(numbers, 2): # numbers에서 조합으로 2개 뽑기
        answer.append(sum(i)) # answer에 2개 값의 합 추가
    return list(sorted(set(answer))) # answer에서 set으로 바꿔 중복 제거후 오름차순하고 리스트화
  • 점수는 +7