코딩테스트(프로그래머스 1단계(15번: 두개 뽑아서 더하기)

나이브한코딩·2021년 7월 8일
0

코딩테스트

목록 보기
16/33
post-thumbnail

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

제한사항
numbers의 길이는 2 이상 100 이하입니다.
numbers의 모든 수는 0 이상 100 이하입니다.

입출력 예
numbers result
[2,1,3,4,1][2,3,4,5,6,7]
[5,0,2,7][2,5,7,9,12]

입출력 예 설명
입출력 예 #1

2 = 1 + 1 입니다. (1이 numbers에 두 개 있습니다.)
3 = 2 + 1 입니다.
4 = 1 + 3 입니다.
5 = 1 + 4 = 2 + 3 입니다.
6 = 2 + 4 입니다.
7 = 3 + 4 입니다.
따라서 [2,3,4,5,6,7] 을 return 해야 합니다.
입출력 예 #2

2 = 0 + 2 입니다.
5 = 5 + 0 입니다.
7 = 0 + 7 = 5 + 2 입니다.
9 = 2 + 7 입니다.
12 = 5 + 7 입니다.
따라서 [2,5,7,9,12] 를 return 해야 합니다.

나의 풀이 :

from itertools import combinations

def solution(numbers) :
    answer = set()

    for i in list(combinations(numbers,2)) : # 2개씩 뽑아 중복 없는 조합 생성 
        answer.add(sum(i)) # 함수에 담고, 
    return sorted(answer) # 오름차순 sort

다른 사람 풀이 :

def solution(numbers):
    answer = []
    for i in range(len(numbers)):
        for j in range(i+1, len(numbers)):
            answer.append(numbers[i] + numbers[j])
    return sorted(list(set(answer)))

TAKEAWAY :

  • combination << 이중포문?? 뭐가 더 나을까.

느낀점 :

  • 이중 포문 다루는 법도 익혀보자!
profile
안녕하세요, 코딩 공부하는 비전공자 취준생입니다.

0개의 댓글

관련 채용 정보