티스토리 뷰

728x90
반응형

 

https://programmers.co.kr/learn/courses/30/lessons/68644

 

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

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

programmers.co.kr

 

 


 

 

<나의 풀이>

# 조합함수를 사용하기 위한 모듈 추가
from itertools import combinations

def solution(numbers):
    answer = []
    numbers.sort()

    for n in numbers:
        i = numbers.index(n)
        new_list = numbers[:i] + numbers[i+1:] # n을 제외하고 새로운 리스트에 추가
        combin = list(combinations(new_list, 2)) # 조합 함수를 통해 숫자들의 조합 만들기
        for a, b in combin:
            if (a+b) not in answer: # 각 숫자의 조합의 합이 answer에 없으면 추가
                answer.append(a+b)

    answer.sort()
    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)))

  
  
# <배운점> 
# => set 함수로 중복 제거한 점
728x90
반응형
댓글