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

yewon Lee·2023년 3월 22일
0
post-thumbnail

😎 코딩테스트 연습>월간 코드 챌린지 시즌1>두 개 뽑아서 더하기



🤷‍♀️정답이 아닌 코드

import itertools

def solution(numbers):
    nCr = itertools.combinations(numbers, 2)
    print(list(nCr)) #이 코드를 실행하면 왜 for문이 안될까?
    
    answer = []
    for i in list(nCr):
        print(i[0], i[1])
        a = (lambda x,y: x+y)(i[0], i[1])
        answer.append(a)
    
    answer = list(set(answer))

    return answer


print라인을 지우면 예제 케이스는 잘 돌아갔지만 틀린 테스트도 있어 정답은 아닌 코드






👍 수정한 코드


테스트에서 실패가 있던건 sort를 하지 않아서 생긴 문제였다.
set을 하면 sort도 같이 되는 줄 알았는데 아니었나보다.
import itertools

def solution(numbers):
    nCr = itertools.combinations(numbers, 2)
    
    answer = []
    for i in list(nCr):
        a = (lambda x,y: x+y)(i[0], i[1])
        answer.append(a)
    
    return sorted(list(set(answer)))


하지만 print의 오류는 여전히....

profile
시작

0개의 댓글