[코테공부 5/26] 두 개 뽑아서 더하기

km-ji·2025년 5월 26일

프로그래머스

목록 보기
3/3
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 해야 합니다.

풀기 전 정리

  • 일단 오름차순 쓰기 (sort)
  • 중복제거 (set)
  • 이중for문으로 다 더한다음에 리스트에 추가해두고, 중복 제거하기
def solution(numbers):
    answer = []
    
    for i in range(len(answer)):
        answer.append(numbers[i-1]+numbers[i])
    answer = set(answer)
    answer.sort()
    return answer

sort랑 sorted 차이

  • sort 함수는 리스트명.sort( ) 형식으로 "리스트형의 메소드"이며 리스트 원본값을 직접 수정
  • sorted 함수는 sorted( 리스트명 ) 형식으로 "내장 함수"이며 리스트 원본 값은 그대로이고 정렬 값을 반환

피드백

  • 이중 for문을 쓰다가 왜 바꿨냐!!
  • 엥 그리고 길이도 answer라고 썼네ㅋㅋㅋㅋnumbers임
  • sort가 아닌 sorted 쓰기
왜 sort만으로는 불가능한가?
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])
    answer = set(answer)
    answer = sorted(answer)
    return answer

더 나은 풀이

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)))


출처 : 코딩테스트 합격자되기 - 파이썬편

profile
I'm mz. Do you want to try mzing?

0개의 댓글