[백준 13458]시험 감독

문제

총 N개의 시험장이 있고, 각각의 시험장마다 응시자들이 있다. i번 시험장에 있는 응시자의 수는 Ai명이다.

감독관은 총감독관과 부감독관으로 두 종류가 있다. 총감독관은 한 시험장에서 감시할 수 있는 응시자의 수가 B명이고, 부감독관은 한 시험장에서 감시할 수 있는 응시자의 수가 C명이다.

각각의 시험장에 총감독관은 오직 1명만 있어야 하고, 부감독관은 여러 명 있어도 된다.

각 시험장마다 응시생들을 모두 감시해야 한다. 이때, 필요한 감독관 수의 최솟값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 시험장의 개수 N(1 ≤ N ≤ 1,000,000)이 주어진다.

둘째 줄에는 각 시험장에 있는 응시자의 수 Ai (1 ≤ Ai ≤ 1,000,000)가 주어진다.

셋째 줄에는 B와 C가 주어진다. (1 ≤ B, C ≤ 1,000,000)

출력

각 시험장마다 응시생을 모두 감독하기 위해 필요한 감독관의 최소 수를 출력한다.

 

 

---

시간 초과...

import java.util.*;

public class Main {

    public static int solution(int N, int[] A, int B, int C) {
        int sumB = 0; // 오직 1명
        int sumC = 0;
        int answer = 0;

        int[] arrC = new int[N];
        for (int i = 0; i<A.length; i++) {
            int people = A[i];
            people = people - B; // 총감독관이 감시하는 응시자 수 제외
            sumB += 1; // 총감독관 무조건 1명은 있음

            if (people>0) {
                for (int k=1; ; k++) {
                    if (C*k>=people) {
                        sumC += k;
                        break;
                    }
                }
            }
            answer += sumB + sumC; // 총 감독관 sum
            sumB = 0; sumC = 0; // 초기화
        }
        return answer;
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt(); // 시험장 개수
        int[] A = new int[N]; // 각 시험장에 있는 응시자의 수
        for (int i=0; i<N; i++) {
            A[i] = scanner.nextInt();
        }
        int B = scanner.nextInt(); // 총감독관이 한 시험장에서 감시할 수 있는 응시자 수
        int C = scanner.nextInt(); // 부감독관이 한 시험장에서 감시할 수 있는 응시자 수

        System.out.println(solution(N,A,B,C));
    }

}

 

 

Greedy 알고리즘!

자료형을 long이 아닌 int로 하면 오답처리 된다.

import java.util.*;

public class Main {

    public static long solution(int N, int[] A, int B, int C) {
        long answer = 0; // 총 감독관 sum

        for (int a: A) {
            long sumB = 0; // 한 시험장 당 필요한 총감독관 수(always 1)
            long sumC = 0; // 한 시험장 당 필요한 부감독관 수

            sumB = 1;
            if ((a - B) > 0) { // 남은 응시자가 0명이면 부감독관 필요 없음
                if ((a - B) < C) { // 남은 응시자 보다 부감독이 감시 가능한 응시자 수가 큰 경우
                    sumC = 1;
                } else {
                    if ((a - B) % C != 0) { // (응시자-총감독관 감시 가능 응시자 수)/부감독관 감시 가능 응시자 수 <- 0으로 떨어지지 않으면
                        sumC = (a - B) / C +1;
                    } else {
                        sumC = (a - B) / C;
                    }
                }
            }
            answer += sumB + sumC;
        }

        return answer;
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt(); // 시험장 개수
        int[] A = new int[N]; // 각 시험장에 있는 응시자의 수
        for (int i=0; i<N; i++) {
            A[i] = scanner.nextInt();
        }
        int B = scanner.nextInt(); // 총감독관이 한 시험장에서 감시할 수 있는 응시자 수
        int C = scanner.nextInt(); // 부감독관이 한 시험장에서 감시할 수 있는 응시자 수

        System.out.println(solution(N,A,B,C));
    }

}