[Programmers] Lv1_예산 (JAVA)

💻JAVA 코드 바로보기

 

 

□문제 설명

S사에서는 각 부서에 필요한 물품을 지원해 주기 위해 부서별로 물품을 구매하는데 필요한 금액을 조사했습니다. 그러나, 전체 예산이 정해져 있기 때문에 모든 부서의 물품을 구매해 줄 수는 없습니다. 그래서 최대한 많은 부서의 물품을 구매해 줄 수 있도록 하려고 합니다.

물품을 구매해 줄 때는 각 부서가 신청한 금액만큼을 모두 지원해 줘야 합니다. 예를 들어 1,000원을 신청한 부서에는 정확히 1,000원을 지원해야 하며, 1,000원보다 적은 금액을 지원해 줄 수는 없습니다.

부서별로 신청한 금액이 들어있는 배열 d와 예산 budget이 매개변수로 주어질 때, 최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성해주세요.

 

---

 

□제한사항

• d는 부서별로 신청한 금액이 들어있는 배열이며, 길이(전체 부서의 개수)는 1 이상 100 이하입니다.
• d의 각 원소는 부서별로 신청한 금액을 나타내며, 부서별 신청 금액은 1 이상 100,000 이하의 자연수입니다.
• budget은 예산을 나타내며, 1 이상 10,000,000 이하의 자연수입니다.

---

□입출력 예

d	budget	result
[1,3,2,5,4]	9	3
[2,2,3,3]	10	4

 

 

 

□입출력 예 설명

입출력 예 #1
각 부서에서 [1원, 3원, 2원, 5원, 4원]만큼의 금액을 신청했습니다. 만약에, 1원, 2원, 4원을 신청한 부서의 물품을 구매해주면 예산 9원에서 7원이 소비되어 2원이 남습니다. 항상 정확히 신청한 금액만큼 지원해 줘야 하므로 남은 2원으로 나머지 부서를 지원해 주지 않습니다. 위 방법 외에 3개 부서를 지원해 줄 방법들은 다음과 같습니다.

• 1원, 2원, 3원을 신청한 부서의 물품을 구매해주려면 6원이 필요합니다.
• 1원, 2원, 5원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다.
• 1원, 3원, 4원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다.
• 1원, 3원, 5원을 신청한 부서의 물품을 구매해주려면 9원이 필요합니다.

3개 부서보다 더 많은 부서의 물품을 구매해 줄 수는 없으므로 최대 3개 부서의 물품을 구매해 줄 수 있습니다.

입출력 예 #2
모든 부서의 물품을 구매해주면 10원이 됩니다. 따라서 최대 4개 부서의 물품을 구매해 줄 수 있습니다.

 

 

 

---

🌼JAVA 알고리즘

 

□초기화면

 

 

□풀이과정

for문으로 누적값이 예산보다 크지 않을 때까지만 더하고, 초과하면 그때의 i값이 곧 answer값이 되지 않을까? 생각하여 풀었다. 여기서 if문에 sum>=budget이 아닌 sum>budget를 쓴 이유는 i값이 곧 인덱스이기 때문에 -1이 자동으로 적용되도록 하기 위함이다.

class Solution {
    public int solution(int[] d, int budget) {
        int answer = 0;
        int sum = 0;
        for(int i=0;i<d.length;i++){
            sum+=d[i];
            if(sum>budget){
                answer = i;
                break;
            } 
        }
        
        return answer;
    }
}

결과값이 0인것으로 보아, for문에서 if문을 통과하지 않았음을 알 수 있다.

위의 방식으로 풀었을 때 case2가 맞지 않았는데, case2의 배열값의 총 합은 10으로 예산과 같다. => 총 합이 예산과 같을 때의 경우를 하나 추가했다.

import java.util.*;
class Solution {
    public int solution(int[] d, int budget) {
        int answer = 0;
        int sum = 0;
        
        for(int i=0;i<d.length;i++){
            sum+=d[i];
            if(sum>budget){
                System.out.println(sum);
                System.out.println(i);
                answer = i;
                break;
            }     
        }
        
        if(sum<=budget) answer = d.length;
        
        return answer;
    }
}

그럼에도 테스트 결과가 100점이 나오지 않았는데, 문제에 제시되어있는 모든 예시가 이미 오름차순으로 정렬되어있다는 것을 알았다..! 그래서 Arrays.sort();를 추가하여 배열 d를 오름차순으로 정렬해놓고 진행하니 해결되었다.

 

 

 

 

 

 

 

 

 

 

 

□완성코드

import java.util.*;
class Solution {
    public int solution(int[] d, int budget) {
        int answer = 0;
        int sum = 0;
        
        Arrays.sort(d);
        
        for(int i=0;i<d.length;i++){
            sum+=d[i];
            if(sum>budget){
                System.out.println(sum);
                System.out.println(i);
                answer = i;
                break;
            }     
        }
        
        if(sum<=budget) answer = d.length;
        
        return answer;
    }
}

 

 

 

 

 

 

 

 

---

□링크

코딩테스트 연습 - 예산

 

About Me