[알고리즘] 1931번

오늘 풀어볼 문제는 백준 1931번 문제 "회의실 배정" 이다.

이 문제는 실버1 문제이고 그리디 문제이다.

문제

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

입력

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

출력

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

📌첫 번째 도전📌
우선 순위 큐를 활용해서 문제를 해결하자고 생각했다. 또한 Room이라는 객체를 만들어서 객체 안에 start와 end 값을 넣어줬다.

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());

        PriorityQueue<Room> pq = new PriorityQueue<>(new Comparator<Room>() {
            @Override
            public int compare(Room r1, Room r2) {
                return r1.end - r2.end;
            }
        });

        for(int i=0; i<N; i++) {
            st = new StringTokenizer(br.readLine());
            int start = Integer.parseInt(st.nextToken());
            int end = Integer.parseInt(st.nextToken());
            Room room = new Room(start, end);
            pq.add(room);
        }

        int endTime = 0;
        int count = 0;

        while (!pq.isEmpty()) {
            Room room = pq.poll();
            if (room.start >= endTime) {
                endTime = room.end;
                count++;
            }
        }
        System.out.println(count);
    }

    public static class Room {
        private int start;
        private int end;

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

하지만.. 실패가 떴다 왜일까??!??!!?!?

📌두 번째 도전📌
종료 시간이 같은 회의일 경우를 생각을 못 했다. 종료 시간이 같을 경우에는 시작 시간이 빠른 애를 우선적으로 해야 한다는 로직을 빼버렸다. 그래서 이를 추가해서 다시 코드를 돌렸다!!

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());

        PriorityQueue<Room> pq = new PriorityQueue<>(new Comparator<Room>() {
            @Override
            public int compare(Room r1, Room r2) {
                if(r1.end == r2.end) {
                    return r1.start - r2.start;
                }
                return r1.end - r2.end;
            }
        });

        for(int i=0; i<N; i++) {
            st = new StringTokenizer(br.readLine());
            int start = Integer.parseInt(st.nextToken());
            int end = Integer.parseInt(st.nextToken());
            Room room = new Room(start, end);
            pq.add(room);
        }

        int endTime = 0;
        int count = 0;

        while (!pq.isEmpty()) {
            Room room = pq.poll();
            if (room.start >= endTime) {
                endTime = room.end;
                count++;
            }
        }
        System.out.println(count);
    }

    public static class Room {
        private int start;
        private int end;

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

[문제 출처] : https://www.acmicpc.net/problem/1931