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

iamjinseo·2022년 8월 18일
0

문제풀이-Python

목록 보기
57/134
post-custom-banner

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

제한사항

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

풀이

from itertools import combinations as co
def solution(numbers):
    answer = []
    
    res = co(numbers, 2) # NC2
    
    for a, b in res:
        answer.append(a+b) # 두 조합의 합 넣기
        
    return sorted(set(answer))

보자마자 조합 생각남.
두 개로 이루어진 조합 만들어 합 구하고 중복제거한다음 오름차순 정렬함.

profile
일단 뭐라도 해보는 중
post-custom-banner

0개의 댓글