728x90
문제
https://school.programmers.co.kr/learn/courses/30/lessons/82612
풀이
Level 1 문제
💡 List Comprehension 활용
List Comprehension을 사용해서 간단하게 필요한 List를 생성해서 계산해주었다.
💡sum()
List Comprehension과 sum()을 활용하여 코드의 길이를 줄였음 !
List Comprehension이 궁금하다면?
https://seokmimmmmmmmm.tistory.com/244
코드
def solution(price, money, count):
answer = sum([price * i for i in range(1, count + 1)]) - money
if answer < 0:
return 0
else:
return answer
728x90