[프로그래머스] [Java] 부족한 금액 계산하기

안녕하세요.

오늘은 프로그래머스 코딩테스트 연습 문제인 "부족한 금액 계산하기" 을 풀어 보았습니다.

 

문제 링크

---

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

 

 

 

문제 풀이 방법

---

이용료의 총 합을 수학식으로 풀어내면 price * (1 ~ count의 합) 이였습니다.

 

저는 1 ~ count까지의 합을 구하는 로직을 반복분을 사용하지 않고 수식을 이용하여 빠르게 구했습니다.

1 ~ n = (n*(1+n)) / 2

(참고 링크: https://en.wikipedia.org/wiki/1_%2B_2_%2B_3_%2B_4_%2B_%E2%8B%AF)

 

하지만 이것으로는 부족했습니다.

 

price의 최대값은 2500, count의 최대값 또한 2500이라 두 최대값을 넣고 테스트를 진행하면 int의 최대범위를 넘어 overflow가 나는 것이였죠.

 

그래서 총 비용을 구하는 로직에서 고의로 long으로 형 변환을 일으켜 계산되는 값이 long으로 나오도록 하였습니다.

 

 

이러한 방법을 사용하여 저는 문제를 풀었고,

이번 문제는 평균 약 0.02ms 정도의 속도로 테스트를 통과하였습니다.

코드

---

class Solution {
    public long solution(int price, int money, int count) {
        long result = price * ((long)count * (1 + count) / 2);
        
        if(result <= money) {
            return 0;
        }
        else 
        {
            return result - (long)money;    
        }
    }
}