[Programmers] 두 개 뽑아 더하기

KyeongHun Kim·2024년 1월 10일

Coding Test 연습

목록 보기
6/14
post-thumbnail

문제

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

제한사항

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

풀이

  1. 두개의 수를 뽑는 경우의 수를 구한다.
  2. 그것을 더하고 중복되는 경우는 리스트에 추가하지 않는다.
from itertools import combinations
def solution(numbers):
    answer = []
    for i in combinations(numbers, 2): # 두 개의 수를 뽑아 조합을 만든다.
        if sum(i) not in answer: # 중복된 값을 추가하지 않는다.
            answer.append(sum(i))
    return sorted(answer)
profile
기본에 충실하자!

0개의 댓글