Python/CoTe

[programmers] 부족한 금액 계산하기

joannekim0420 2022. 2. 26. 18:41
728x90

https://programmers.co.kr/learn/courses/30/lessons/82612

 

코딩테스트 연습 - 부족한 금액 계산하기

새로 생긴 놀이기구는 인기가 매우 많아 줄이 끊이질 않습니다. 이 놀이기구의 원래 이용료는 price원 인데, 놀이기구를 N 번 째 이용한다면 원래 이용료의 N배를 받기로 하였습니다. 즉, 처음 이

programmers.co.kr

 

 

 

def solution(price, money, count):
    answer = sum(price*i for i in range(1, count+1)) - money
    return answer if answer > 0 else 0

포인트!

1. sum, for 한 라인으로 쓰기

2. return if 같은 라인에 쓰기

    return value if (조건) else value2