[코딩테스트] [1931] 회의실 배정

[1931] 회의실 배정

시간 제한 메모리 제한 제출 정답 맞힌 사람 정답 비율

2 초

128 MB

237241

79093

54552

30.999%

문제

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

입력

첫째 줄에 회의의 수 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N+1 줄까지 각 회의의 정보가 주어지는데 이것은 공백을 사이에 두고 회의의 시작시간과 끝나는 시간이 주어진다. 시작 시간과 끝나는 시간은 231-1보다 작거나 같은 자연수 또는 0이다.

출력

첫째 줄에 최대 사용할 수 있는 회의의 최대 개수를 출력한다.

처음 문제 풀이

이 문제를 보고 차분하게 문제를 이해하기 위해 정리하였다.

1개의 회의실이 있고 회의의 총 개수 N개가 있다. 각각 회의 I는 시작 시간과 끝 시간이 다른데 이 때 최대한 회의를 많이 사용한다면 최대 몇개의 회의를 할 수 있는지 찾는 문제이다.

문제는 이해했고 한 번 예시를 대입해보자.

일단 시작시간이 같은 회의는 병합하자.

회의 시작시간이 같은 경우 총 회의 시간이 더 작은 경우가 유리하니 총 회의시간이 작은 쪽으로 중복 제거를 하겠다. 그래서 TreeMap을 사용했다.

if문을 사용해서 key가 있을경우 value값이 더 작은 경우만 들어가게 하였다.

        Map<Integer, Integer> map = new TreeMap<>();

        for(int i = 0; i<N; i++) {
            int key = sc.nextInt();
            int value = sc.nextInt();
            if(map.containsKey(key)) {
                if(map.get(key) > value)
                    map.put(key,value);
            } else {
                map.put(key,value);
            }

총 회의시간을 기준으로 정렬하자.

총 회의시간이 작은 것들이 유리하다고 생각하여 총 회의시간을 기준으로 정렬을 해보겠다.

Array정렬을 하기 위해 entryArray를 만들어서 totalTime으로 오름차순 정렬을 실행했다.

        Map.Entry<Integer, Integer>[] entryArray = map.entrySet().toArray(new Map.Entry[0]);

        // totalTime을 기준으로 오름차순 정렬
        Arrays.sort(entryArray, (o1, o2) -> {
            int totalTime1 = o1.getKey() - o1.getValue();
            int totalTime2 = o2.getKey() - o2.getValue();
            return Integer.compare(totalTime1, totalTime2); // totalTime을 비교
        });

하지만 이중반복문으로 비교를 해야한다?

이렇게 해서 조건을 만족하는 개수를 구하려고 하는데

조건은 끝 시간(i) ≤ 시작시간(j) 이거나 시작시간(i) > 시작시간(j) && 시작(i) ≥ 끝(j)이다.

이러면 이중반복문을 사용해서 하나하나씩 모두 비교를 해주어야 하는데 시간초과가 발생할 것이다..😭

다른 풀이를 해보자

정렬 기준을 다르게 해보자.

끝 시간을 기준으로 정렬을 하여 끝시간과 그 다음 회의의 첫 시간을 비교하여 입력해줬다.

그런데 이 문제를 총 3시간을 하면서 깨달았다.

이 문제는 Map을 사용해서 풀면 안된다..

예제가 하나라서 도대체 왜 틀렸을까를 2시간을 넘게 하면서 깨달았다..

모든 조건을 충족 시키려면 Map으로 중복을 제거하는 것이 아닌 이중 배열로 모든 케이스를 받아서 비교를 해야하는 것..… 코테 어렵다.. 🫠

알고리즘을 구현할때는 차분하게 모든 조건을 다시 한 번 계산하면서 해보자.

나는 성격이 일단 부딪혀보는 성격이기 때문에 모든 포기하지 않고 부딪혀보는 성격이다.

하지만 이것은 코테하면서 좋지 않다.. 모든 문제는 완전탐색이 아니기 때문. BruteForce는 좋지 않다.

Map을 사용한 코드 (틀림)

아래 코드는 Key값이 없으면 값을 넣어주고 존재한다면 Value가 더 작은 쪽을 넣어주는 코드이다.

문제를 아래 코드로 풀었을 때 예제1를 모두 충족시키기 때문에 제출을 하였지만 틀렸고

그 이후 4,4 / 5,5 등 시작과 종료가 같은것은 무조건 추가를 해주면 되겠다고 생각하여 추가해주었지만

다른 회의 시간이 4~6일 경우, 5,5가 있으면 회의를 할 수 없기에 5,5도 있어야한다.

즉 이 문제는 Map을 사용하면 안된다. Map은 Key와 대응되는 Value가 필요할 때 사용하자.

이 문제에서는 Key와 대응하긴 하지만 연관이 크게 필요한 경우가 아니기에 이차원 배열로 하는 것이 더 맞고 애초에 문제자체도 Map으로 더 조건을 많이 생각해야 한다.

import java.util.*;

class Main
{
    public static void main (String[] args) {
        Scanner sc = new Scanner(System.in);

        int N = sc.nextInt();

        Map<Integer, Integer> map = new TreeMap<>();

        for(int i = 0; i<N; i++) {
            int key = sc.nextInt();
            int value = sc.nextInt();
            if(map.containsKey(key)) {
                if(map.get(key) > value)
                    map.put(key,value);
            } else {
                map.put(key,value);
            }
        }

        Map.Entry<Integer, Integer>[] endArray = map.entrySet().toArray(new Map.Entry[0]);

        Arrays.sort(endArray, (o1, o2) -> {
            return o1.getValue() - o2.getValue();
        });

        Map.Entry<Integer, Integer>[] answer = new Map.Entry[endArray.length];

        if (endArray.length > 0) {
            answer[0] = endArray[0];
        }

        int count = 1;

        for(int i=0; i<endArray.length; i++){

            if(answer[0].getValue() <= endArray[i].getKey()) {
                answer[0] = endArray[i];
                count++;
            } else {

            }
        }

        System.out.println(count);
    }
}

이차원 Array로 푼 문제풀이

시작과 끝값을 이차원 배열에 입력해주고 정렬 기준을 끝 시간을 기준으로 정렬 하자.

그 이유는 시작으로 한다면 언제 끝나는지 모르기 때문에 결국 끝을 알아야 하고

끝을 기준으로 한다면 끝이 나면 회의실은 빈상태이기에 그 다음시간이 들어올 수 있다.

만약 끝시간이 같을 경우에는? 시작시간을 오름차순으로 정렬하여 풀자.

그 이유는 시작 시간이 더 이른 회의가 오도록하여 선택할 수 있는 회의의 수를 더 최대화할 수 있다.

import java.util.*;

class Main
{
    public static void main (String[] args) {
        Scanner sc = new Scanner(System.in);

        int N = sc.nextInt();

        Integer[][] inputs = new Integer[N][2];

        for(int i=0; i<N; i++) {
            int start = sc.nextInt();
            int end = sc.nextInt();

            inputs[i][0] = start;
            inputs[i][1] = end;
        }
        
        Arrays.sort(inputs, (o1, o2) -> {
            if(o1[1] == o2[1])
                return o1[0] - o2[0];
            return o1[1] - o2[1];
        });

        int count = 0;
        int lastEndTime = -1; // 마지막 회의 종료 시간

        for (Integer[] entry : inputs) {
            int startTime = entry[0];
            int endTime = entry[1];

            if (lastEndTime <= startTime) {
                count++; // 회의 수 증가
                lastEndTime = endTime; // 현재 회의의 종료 시간 업데이트
            }
        }
        
        System.out.println(count);
    }
}