두 개 뽑아서 더하기

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

프로그래머스

문제 설명

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

 

제한사항

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

---

code

# Lv.1
# 두 개 뽑아서 더하기
def solution(numbers):
    answer = []
    for i in range(len(numbers)):
        for j in range(i + 1, len(numbers)):
            num = numbers[i] + numbers[j]
            if num not in answer:
                answer.append(num)
    return sorted(answer)

https://github.com/soaringwave/Python-algorithm-studying/blob/main/programmers/Lv.1/sum_combination.py

 

알고리즘

반복문을 통해 중복없이 두 수를 골라 더한 값을 num에 저장하고, num의 값이 answer 배열에 없다면(중복 방지) answer 배열에 추가한다. 그리고 오름차순으로 정렬된 answer를 반환한다.

 

피드백

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

다른 사람의 풀이 중 하나다. 이 코드를 보면 우선 combiation을 통해 두 수가 짝인 원소를 지닌 리스트를 만든다. 그리고 이 두 수의 짝을 더한 값들을 set을 통해 중복을 없애고, sorted()를 통해 오름차순으로 정렬해 반환한다. 가독성도 괜찮고, 코드를 더 줄일 수 있는 방법이라서 기록한다.

 

set: {} 집합

https://velog.io/@insutance/Python-set-%EC%9D%B4%EB%9E%80

[Python] set() 이란