Level 1. 두개 뽑아서 더하기

18. 두개 뽑아서 더하기

코딩테스트 연습 > 월간 코드 챌린지 시즌1 > 두 개 뽑아서 더하기
https://programmers.co.kr/learn/courses/30/lessons/68644

문제 정리

Input value =

• number: 정수 리스트

1. 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더하기

2-1. itertools의 combination 라이브러리를 이용한 방법

2-2. 이중 for문을 이용한 방법

return list(sum(a,b))

제한 조건

ㆍ numbers의 길이는 2 이상 100 이하입니다.

• numbers의 모든 수는 0 이상 100 이하입니다.

풀이 방법

이중 for 문 이용

def solution(numbers):
    answer = set()
    for i in range(len(numbers)):
        for j in range(i+1, len(numbers)):
            answer.add(numbers[i] + numbers[j])
    answer = list(answer)
    answer.sort()
    return answer

# Code test
numbers = [2,1,3,4,1]
solution(numbers)

itertools-combinations 라이브러리 이용

from itertools import combinations

def solution(numbers):
    list_num = list(combinations(numbers,2))
    answer = set()
    for n in list_num:
        answer.add(n[0]+n[1])
    return list(sorted(answer))

# Code test
numbers = [2,1,3,4,1]
solution(numbers)