오늘의 코딩 테스트(부족한 금액 계산하기)

새로 생긴 놀이기구는 인기가 매우 많아 줄이 끊이질 않습니다. 이 놀이기구의 원래 이용료는 price원인데, 놀이기구를 N번 째 이용한다면 원래 이용료의 N배를 받기로 하였습니다. 즉, 처음 이용료가 100이었다면 2번째에서는 200, 3번째에서는 300으로 요금이 인상됩니다.

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

 

제한사항

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

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

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

 

 

내가 푼 풀이

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

        return answer;
    }
}

 

횟수만큼 for문을 돌려 total을 구하여 값을 비교하여 풀었다.

동일하게 for문과 if문을 사용하지만 for문에서 total을 구하는 것이 아닌 money에서 가격 * 횟수를 빼주는 방법도 있다.

 

 

횟수만큼 반복문을 돌려 지닌 돈에서 가격 * 횟수만큼 뺀 뒤 if 문에서 조건에 맞게 답을 도출하는 풀이

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

 

등차수열의 합과 Math.max() 함수를 활용하는 풀이

class Solution {
    public long solution(int price, int money, int count) {
        
        long answer = 0;
        
        long total = (long)price * count * (count + 1) / 2;

        return Math.max(0, total - money);
    }
}

 

등차수열을 이용하여 total을 구하고 Math.max() 함수를 활용하여 0과 total - money의 값을 비교하여 큰 값을 리턴 한다.

 

 

IntStream과 삼항 연산자를 활용하는 풀이

class Solution {
    public long solution(int price, int money, int count) {        
        
        long sum = IntStream
            .rangeClosed(1,count)
            .mapToLong(i-> i*price)
            .sum();
        
        
        long answer = (money > sum) ?  0 : sum - money ;      
        
        return answer;
    }
}