자바랑 코덕코덕

[문제]

한 개의 회의실이 있는데 이를 사용하고자 하는 N개의 회의에 대하여 회의실 사용표를 만들려고 한다. 각 회의 I에 대해 시작시간과 끝나는 시간이 주어져 있고, 각 회의가 겹치지 않게 하면서 회의실을 사용할 수 있는 회의의 최대 개수를 찾아보자. 단, 회의는 한번 시작하면 중간에 중단될 수 없으며 한 회의가 끝나는 것과 동시에 다음 회의가 시작될 수 있다. 회의의 시작시간과 끝나는 시간이 같을 수도 있다. 이 경우에는 시작하자마자 끝나는 것으로 생각하면 된다.

[문제 풀이]

이 문제는 회의가 끝나는 시간을 기준으로 오름차순 정렬을 해야한다.

끝나는 시간이 같다면 시작하는 시간이 작은 것을 우선으로 정렬한다.

차례대로 회의실 시간이 끝나는 시간이 다음 회의 시작 시간보다 작거나 같으면 회의를 진행한다. (cnt+1)

그리고 다음 회의 시작 가능 시간을 그 회의 끝나는 시간으로 바꿔준다.

그래서 회의 가능한 횟수 cnt를 출력해 주면 되는 문제이다.

[코드]

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main{

    static int N;
    static Meeting[] meetings;

    static int stoi(String s) {
        return Integer.parseInt(s);
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = stoi(br.readLine());
        meetings = new Meeting[N];

        for (int i=0; i<N; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            meetings[i] = new Meeting(stoi(st.nextToken()), stoi(st.nextToken()));
        }
        Arrays.sort(meetings);

        System.out.println(getRoom());
    }

    static int getRoom() {
        int cnt = 1;
        int max = 1;

        int cur = meetings[0].end;

        for (int i=1; i<N; i++) {
            if (cur <= meetings[i].start) {
                cnt++;
                cur = meetings[i].end;
            }
            max = Math.max(cnt, max);
        }

        return max;
    }

    static class Meeting implements Comparable<Meeting>{
        int start;
        int end;

        public Meeting(int start, int end) {
            this.start = start;
            this.end = end;
        }

        @Override
        public int compareTo(Meeting o) {
            if (this.end == o.end) {
                return this.start-o.start;
            }
            return this.end-o.end;
        }
    }
}

[실행 결과]

결과는 위와 같다.

[문제]

인하은행에는 ATM이 1대밖에 없다. 지금 이 ATM앞에 N명의 사람들이 줄을 서있다. 사람은 1번부터 N번까지 번호가 매겨져 있으며, i번 사람이 돈을 인출하는데 걸리는 시간은 Pi분이다.

사람들이 줄을 서는 순서에 따라서, 돈을 인출하는데 필요한 시간의 합이 달라지게 된다. 예를 들어, 총 5명이 있고, P1 = 3, P2 = 1, P3 = 4, P4 = 3, P5 = 2 인 경우를 생각해보자. [1, 2, 3, 4, 5] 순서로 줄을 선다면, 1번 사람은 3분만에 돈을 뽑을 수 있다. 2번 사람은 1번 사람이 돈을 뽑을 때 까지 기다려야 하기 때문에, 3+1 = 4분이 걸리게 된다. 3번 사람은 1번, 2번 사람이 돈을 뽑을 때까지 기다려야 하기 때문에, 총 3+1+4 = 8분이 필요하게 된다. 4번 사람은 3+1+4+3 = 11분, 5번 사람은 3+1+4+3+2 = 13분이 걸리게 된다. 이 경우에 각 사람이 돈을 인출하는데 필요한 시간의 합은 3+4+8+11+13 = 39분이 된다.

줄을 [2, 5, 1, 4, 3] 순서로 줄을 서면, 2번 사람은 1분만에, 5번 사람은 1+2 = 3분, 1번 사람은 1+2+3 = 6분, 4번 사람은 1+2+3+3 = 9분, 3번 사람은 1+2+3+3+4 = 13분이 걸리게 된다. 각 사람이 돈을 인출하는데 필요한 시간의 합은 1+3+6+9+13 = 32분이다. 이 방법보다 더 필요한 시간의 합을 최소로 만들 수는 없다.

줄을 서 있는 사람의 수 N과 각 사람이 돈을 인출하는데 걸리는 시간 Pi가 주어졌을 때, 각 사람이 돈을 인출하는데 필요한 시간의 합의 최솟값을 구하는 프로그램을 작성하시오.

[문제 풀이]

입력을 배열로 받고 오름차순으로 정렬한다. 1 2 3 4 5 라고 하면

각각의 대기 시간은 아래와 같다.

1번째 사람: 1

2번째 사람: 1+2

3번째 사람: 1+2+3

4번째 사람: 1+2+3+4

5번째 사람: 1+2+3+4+5

따라서 1*5 + 2*4 + 3*3 + 4*2 + 5 한 결과가 답이 된다.

코드로 표현하면 아래와 같다.

[코드]

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
    static int N;
    static int[] arr;

    static int stoi(String s) {
        return Integer.parseInt(s);
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = stoi(br.readLine());

        arr = new int[N];
        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i=0; i<N; i++) {
            arr[i] = stoi(st.nextToken());
        }

        Arrays.sort(arr);

        int ans = 0;
        int n = N;
        for (int i=0; i<N; i++, --n) {
            ans += arr[i] * n;
        }
        System.out.println(ans);
    }
}

[실행 결과]

실행 결과는 위와 같다.