[Programmers][Python][Level 1] 두 개 뽑아서 더하기

프로그래머스의 Level 01 두 개 뽑아서 더하기 문제입니다.

 

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

 

문제

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

 

풀이

두 가지를 신경써주시면 됩니다.

 

1. answer 리스트에 중복되지 않는 서로 다른 인덱스 numbers의 값들을 추가해줍니다.

2. answer 리스트를 오름차순 정렬을 해주면 됩니다.

 

첫 번째 풀이

1. numbers인 [2,1,3,4,1] 각각의 항목을 더할 수 있게 for문을 돌려줍니다.
2. numbers의 각각의 더한 값들을 answer리스트에 담아줍니다.
   
   • answer.append(numbers[i] + numbers[j])
3. answer 리스트에 중복되는 값들을 set() 집합으로 없애주고 오름차순 정렬을 해줍니다.
   • sorted(set(answer))
   • sorted()는 정렬된 리스트를 반환해줍니다.

 

def solution(numbers):

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

 

 

다른 사람의 풀이 - combinations 사용

다른 분은 combinatons(조합)를 사용해서 해결했습니다.

• combinations(iterable, r) : 사용자의 입력값인 iterable에서 r의 길이의 조합을 리턴해줍니다.
  • combinations('ABCD', 2) : AB AC AD BC BD CD
  • python docs combinatinos => docs.python.org/ko/3/library/itertools.html#itertools.combinations

 

from itertools import combinations

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

    for i in l:
        answer.append(i[0]+i[1])
 
    return sorted(set(answer))