[Python] 프로그래머스 - 두 개 뽑아서 더하기

오늘부터 Python으로 알고리즘 테스트를 준비할 예정기에 하루에 하나씩 글을 작성하는 것을 목표로 남은 2월달을 보내보자.

 

일단 기초 문법도 모르는 상태이기 때문에 구글링을 하면서 프로그래머스 문제를 풀어보고 해당 문제에서 나온 문법과 알고리즘에 대해서 정리할 계획이다.

 

두 개 뽑아서 더하기

해당 문제는 주어진 배열 numbers에서 서로 다른 인덱스에 있는 두 개이 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return해주는 문제이다.

 

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

 

일단 2중 for문을 통해 배열을 돌리는 부분부터 보자.

 

for i in range(len(numbers) - 1)

파이썬은 len이라는 함수를 통해 해당 배열의 길이를 구할 수 있기때문에 주어진 배열의 길이 - 1 까지 반복해주고

for i in range(i+1, len(numbers))

i+1부터 배열의 길이까지 2중으로 반복해준다.

 

그 뒤에 if not in 문을 사용해서 return 해줄 배열에 numbers[i] + numbers[j] 가 존재하지 않는다면 answer배열에 append시켜준다. 

 

그 후 answer배열을 sorted함수를 사용해서 오름차순으로 정렬하여 return을 해주면 된다.

 

※ set함수를 사용하게 되면  if not in구문을 사용하지 않고도 중복제거를 할 수 있게 된다.