[프로그래머스/Programmers] 두 개 뽑아서 더하기 - Python

Link

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

 

Solution

해당 문제는 제한사항이 작기 때문에 이중 for문과 combinations를 모두 사용해도 문제가 없다고 판단하였다.

nums의 길이는 2 이상 100 이하입니다.
nums의 모든 수는 0 이상 100이하입니다.

 

def solution(nums):

    return sorted(set([nums[i] + nums[j] for i in range(len(nums)) for j in range(i+1, len(nums))]))
    
    # sorted(): list 반환

from itertools import combinations

def solution(nums):
    sums = list(combinations(nums, 2))

    return sorted(set([sum(i) for i in sums])) # sorted(): list 반환

🔑 key point: Limits
📌 Tip: set(), itertools.combinations()