코딩테스트/프로그래머스 level 1

[프로그래머스 level_1] 예산 for JAVA

냠냠:) 2020. 5. 12. 14:51

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

[문제 설명]

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개 부서의 물품을 구매해 줄 수 있습니다.

 

[풀이]

현재 예산에서 각 부서에서 신청한 금액에 최대한 맞춰 최대 몇 개의 부서를 지원할 수 있는지 답을 내야 하는 문제입니다.

 

문제에는 살짝 약점? 같은게 있는데 몇 개의 부서를 지원하고 남는 금액에 대해서는 답을 도출하지 않아도 되는 것입니다.

 

즉 입출력 예1을 보면 1,2,3원을 신청한 부서를 지원하면 3원이 남고 1,3,5원을 신청한 부서를 지원하면 0원이 남는데

결국 3개의 부서를 지원하는 것과 똑같습니다.

 

마치 문제는 남은 돈이 0에 가깝게 코딩을 하라는 것 같지만 사실은 그럴 필요가 없습니다. 우리는 남는 금액보다 얼마나 많은 부서를 지원할 수 있는지만 집중하면 됩니다.

 

3의 차이가 나지만 지원한 부서는 동일하게 3곳이라는 겁니다.

 

만약 신청금액이 적은 순서대로 지원하게 되면 어떻게 될까요?

 

만약 신청 금액이 [1.2.3.5.9]이고 지원금액이 9원일 때 이 배열이 오름차순으로 정렬을 한 것이라면

 

우리는 1,2,3 만 지원 할 수 있을 것입니다. 

이미 예산은 3이 남아 있기 때문에 5를 신청한 부서는 지원을 못 받는 겁니다.

 

이런 식으로 배열을 오름차순으로 정렬한 뒤 순차적으로 신청 금액을 더해주고 그 값이 예산을 넘겼을 때 우리는 멈추고 이전까지 세었던 카운터를 반환해주기만 하면 됩니다.

 

[코드]

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import java.util.Arrays;
 
public class Budget {
    public int solution(int[] d, int budget) {
        int sum = 0;
        int count =0 ;
        Arrays.sort(d);
        
        for(int i = 0; i < d.length; i++) {
            if(sum + d[i] > budget) {
                break;
            }
            
            sum += d[i];
            count++;
        }
        return count;
    }
cs

 

 

반응형