코딩테스트/프로그래머스

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

박소민 2022. 3. 17. 14:07
문제) 두 개 뽑아서 더하기
 

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

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

programmers.co.kr

 

  • 내 풀이
    • set 함수: 중복 제거
def solution(numbers):
    answer = []
    for i in range(len(numbers)):
        for j in range(len(numbers)):
            if i==j:
                continue
            answer.append(numbers[i]+numbers[j])
    answer=sorted(set(answer)) #tuple은 중복 가능, set은 중복 불가
    
    return answer

 

  • 다른 사람 풀이 1
    • 안쪽 for문의 범위를 i+1 부터로 하면 앞에서 더했던 값들 끼리 다시 더하지 X 
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=sorted(set(answer))
    
    return answer

 

  • 다른 사람 풀이 2
    • 이중 반복문 한 줄로 작성
    • set()  대신 { } 사용:  중복 제거
def solution(numbers):
    answer=sorted({numbers[i] + numbers[j] for i in range(len(numbers)) for j in range(i+1, len(numbers))})
    
    return answer