최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성해주세요.
테스트 (인텔리제이)
이번에는 다른 사람 풀이 중에서 너무 좋은 로직들이 있어 가져와 봤습니다.
첫 번째 풀이는 제 풀이로 최대한 많은 원소를 더해줘야 하기 때문에 작은 수부터 하나씩 더할 수 있게 d를 오름차순 정렬하였고
정렬된 d의 원소를 하나씩 꺼낼 수 있도록 for문을 사용했습니다.
이후 if문으로 원소들의 합을 저장하기 위해 새롭게 선언한 sum 변수가 budget보다 작거나 같으면 d의 원소를 더해주고 answer를 1씩 증가하게 하여 카운트 해줍니다.
이번에는 또 다른 if문으로 더해준 sum이 budget보다 크면 sum에서 해당 d의 원소를 빼주고 answer를 1 감소하게 하면 해결할 수 있습니다.
두 번째 풀이는 sum 변수 선언이 필요 없습니다.
정렬된 d의 크기만큼 for문이 순회하고 budget에서 해당 d의 원소를 뺴주도록 합니다.
if문으로 butget이 0보다 작아지면 break를 이용하여 for문을 정지하도록 하고
if문의 break 되기 전까지 answer는 1씩 증가하도록 하는 풀이를 가져와 봤습니다.
세 번째 풀이는 너무 신기했던 풀이였습니다.
똑같이 정렬을 하지만
이번에는 while문을 사용하고 내부 로직은 없습니다.
조건을 걸어주는 괄호에 budget에서 d의 원소를 하나씩 꺼내 빼주도록 하고 그 값이 0보다 크거나 같을 때에만 반복하도록 하고 and 조건으로 answer가 1번 반복할 때마다 1씩 증가하도록 해줍니다 그리고 그 값이 d의 크기보다 작을 때에만 반복하도록 하여 내부 로직 없이 푼 풀이를 가져와 봤습니다.
package com.namji.codingtest;
import java.util.Arrays;
public class 예산 {
public static void main(String[] args) {
/*
정수 배열 d와 정수 budget이 매개변수로 주어집니다.
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개 부서의 물품을 구매해 줄 수 있습니다.
*/
/*
d budget result
[1,3,2,5,4] 9 3
[2,2,3,3] 10 4
[1,2,3,4,5,6,7,1] 7 4
*/
// int[] d = {1,2,3,4,5,6,7,1};
// int budget = 7;
// int sum = 0;
// int answer = 0;
// Arrays.sort(d);
//
// for (int i = 0; i < d.length; i++) {
// if (sum <= budget) {
// sum += d[i];
// answer++;
// }
// if (sum > budget) {
// sum -= d[i];
// answer--;
// }
// }
//
// System.out.println(answer);
// int[] d = {1,2,3,4,5,6,7,1};
// int budget = 7;
// int answer = 0;
// Arrays.sort(d);
//
// for (int i = 0; i < d.length; i++) {
// budget -= d[i];
// if (budget < 0) {
// break;
// }
// answer++;
// }
//
// System.out.println(answer);
int[] d = {1,2,3,4,5,6,7,1};
int budget = 7;
int answer = 0;
Arrays.sort(d);
while ((budget -= d[answer]) >= 0 && ++answer < d.length);
System.out.println(answer);
}
}
프로그래머스 코드
// import java.util.*;
// class Solution {
// public int solution(int[] d, int budget) {
// int sum = 0;
// int answer = 0;
// Arrays.sort(d);
// for (int i = 0; i < d.length; i++) {
// if (sum <= budget) {
// sum += d[i];
// answer++;
// }
// if (sum > budget) {
// sum -= d[i];
// answer--;
// }
// }
// return answer;
// }
// }
// import java.util.*;
// class Solution {
// public int solution(int[] d, int budget) {
// int answer = 0;
// Arrays.sort(d);
// for (int i = 0; i < d.length; i++) {
// budget -= d[i];
// if (budget < 0) {
// break;
// }
// answer++;
// }
// return answer;
// }
// }
import java.util.*;
class Solution {
public int solution(int[] d, int budget) {
int answer = 0;
Arrays.sort(d);
while((budget -= d[answer]) >= 0 && ++answer < d.length);
return answer;
}
}
