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

문제 설명

정수 배열 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 해야 합니다.

 

---

내 답안

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

for문을 중첩하여 완전탐색 구조를 만든다!

numbers배열 내 두 요소의 합에 대한 모든 경우의 수를 plus배열에 저장한다.

중복값이 있으면 안되므로! 이때 set 자료형을 이용한다! (검색해서 찾았다 ㅋ..)

1) plus 배열을 set 자료형으로 변화시켜 toset 에 지정하고

2) 다시 toset을 list 화 하여 newlist 에 저장한다.

sorted(list(set(plus)))

이렇게 해도 같은 결과값이 나온다.

(여담인데 다른사람들 풀이를 보면 return 값에 저런 식을 직접 넣는 경우가 많다.. 그러면 식이 훨씬 깔끔해지긴 하는듯?! 나두 해봐야지)

그런데 왜 굳이 set 자료형의 합을 구하지 않고, list -> set -> list 로 바꿔 list 자료형으로 연산해야 할까?

그건.. 모르겠다 set 자료형은 집합 { } 이기 때문에.. 연산이 안되는것 같다 (나중에 찾아보기)

마지막으로 오름차순 배열을 위해 sorted 함수를 활용한다. 

 

다른사람 풀이 #1

from itertools import combinations

def solution(numbers):
    answer = []
    l = list(combinations(numbers, 2))

    for i in l:
        answer.append(i[0]+i[1])
    answer = list(set(answer))
    answer.sort()

    return answer

itertools 라이브러리 내 combination 함수를 이용했다.

combination 은 리스트 내 모든 조합을 구하는 함수며 사용법은 다음과 같다.

from itertools import combination
list(combination(리스트명,n))

이 식을 이용하면 리스트에서 n개의 요소의 합에 대한 모든 조합이 구해진다.

참고로 두개이상의 리스트 내 모든 조합을 구할땐, 동일 라이브러리의 product 함수를 이용하면 된다.

 

이분은 오름차순을 위해 sort를 사용했는데,

sort는 리스트 그 자체를 단순 정렬만 시키고

sorted는 리스트를 정렬하여 새로운 리스트을 반환하는 기능을 가진다.