본문 바로가기
Algorithm/프로그래머스

[프로그래머스] Lv1 - 두 개 뽑아서 더하기 (파이썬)

by 2HS 2022. 3. 19.

https://programmers.co.kr/learn/courses/30/lessons/68644

 

코딩테스트 연습 - 두 개 뽑아서 더하기

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

programmers.co.kr

 

from itertools import combinations as comb


def solution(numbers):
    arr = list(comb(numbers, 2))
    answer = [sum(temp) for temp in arr]
    answer = sorted(list(set(answer)))
    return answer

 

풀이
1. combinations를 이용해서 순서와 상관없이 두 가지를 뽑는 경우의 수를 리스트화 한다.
2. 해당 리스트의 합을 answer리스트로 초기화
3. answer를 집합의 성질을 이용해서 중복값을 제거하고 다시 리스트화 한 후 정렬
4. answer 반환