-
[프로그래머스] 두 개 뽑아서 더하기 (Python)코딩테스트 2021. 3. 27. 01:55
import itertools def solution(numbers): answer = [] temp = [] temp = itertools.combinations(range(len(numbers)), 2) for i in temp: answer.append(numbers[i[0]] + numbers[i[1]]) answer = sorted(list(set(answer))) return answer
itertools의 조합 함수 combinations를 사용해 인덱스를 2개 뽑고, 뽑은 인덱스를 사용해 각 값을 더한 수들을 answer 리스트에 append한 후 중복 제거를 위해 set을 이용한다.
오름차순 정렬이니 sorted를 사용해 정렬한다.
'코딩테스트' 카테고리의 다른 글
[프로그래머스] 2016년 (Python) (0) 2021.03.27 [프로그래머스] 같은 숫자는 싫어 (Python) (0) 2021.03.25 [프로그래머스] 평균 구하기 (Python) (0) 2021.03.25 [프로그래머스] 문자열 내 p와 y의 개수 (Python) (0) 2021.03.25 [프로그래머스] 1878 - 나머지 한 점(Python) (0) 2021.03.06