[프로그래머스] 부족한 금액 계산하기 - 자바(JAVA)

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

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

 

문제 해석

새로 생긴 놀이기구는 인기가 너무 많아 줄이 끊이지 않습니다.

원래 이용료는 price인데 놀이기구를 N번째로 이용한다면 원래 이용료의 N배를 받습니다.

처음 이용료가 100이라면 2번째는 200, 3번째는 300으로 요금이 인상됩니다.

 

놀이기구를 count번 타게 되면 현재 자신이 가지고 있는 금액에서 얼마가 모자라는지를 return 하도록 하세요.

 

문제 풀이 전 설계

count의 범위는 1~2500입니다.

 

반복문 한번만에 문제를 해결할 수 있어 보입니다. O(N)

 

count만큼 반복을 돌며 sum에 이용료를 모두 쌓아놓습니다.

 

이후에 answer = money - sum을 해주고 answer > 0이라면 return 0

그렇게 않으면 -answer을 반환해줍니다.

 

코드

class Solution {
    public long solution(int price, int money, int count) {
        long answer = -1;
        long sum = 0;
        
        for(int i=1; i<= count; i++){
            sum += (price*i);
        }
        
        answer = money - sum;
        if(answer > 0){
            return 0;
        }
        
        return -answer;
    }
}