[그 외]/ㄴ (코테연습 : 파이썬 ver)
[프로그래머스] 두 개 뽑아서 더하기 (파이썬)
__hyeon2__
2021. 6. 19. 17:20
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
반응형