[프로그래머스] Lv.1 부족한 금액 계산하기 java

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

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

 

문제

놀이기구의 원래 이용료는 price원, 놀이기구를 N번째 이용한다면 원래 이용료의 N배를 받기로 함. 즉, 처음 이용료가 100이었다면 2번째는 200, 3번째는 300으로 요금이 인상. 놀이기구를 count번 타게 되면 현재 자신이 가지고 있는 금액에서 얼마가 모자라는지 return하는 함수를 완성하시오. 단, 금액이 부족하지 않으면 0을 return.

제한조건

놀이기구의 이용료 price:1 <= price <= 2,500, price는 자연수

처음 가지고 있던 금액 money : 1 <=  money <= 1,000,000,000, money는 자연수

놀이기구의 이용 횟수 count : 1 <= count <= 2,500, count는 자연수

 

.java

class Solution {
    public long solution(int price, int money, int count) {
        long answer = 0;
        long total=0;
		
        //탄 횟수만큼 금액 늘리며 더하기
        for(int i=1; i<count+1; i++){
            total += price*i;
        }
        //가진 돈보다 비용이 많이 들 경우
        if(money<total)
            answer = total-money;
        //모자라지 않는 경우
        else
            answer = 0;
        
        return answer;
    }
}

실행 결과