백준 - 1931번(회의실 배정)

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

문제

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

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Iterator;
import java.util.StringTokenizer;
import java.util.TreeSet;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(reader.readLine());
        TreeSet<Meeting> treeSet = new TreeSet<>();
        int count = 1;
        while (N-- > 0) {
            StringTokenizer tokenizer = new StringTokenizer(reader.readLine());
            treeSet.add(new Meeting(Integer.parseInt(tokenizer.nextToken()), Integer.parseInt(tokenizer.nextToken())));
        }

        Iterator<Meeting> iterator = treeSet.iterator();
        Meeting temp = iterator.next(); // 초깃값
        while (iterator.hasNext()) {
            Meeting current = iterator.next();
            if (current.start >= temp.end) {
                count++;
                temp = current;
            }
        }

        System.out.println(count);

        reader.close();
    }

}

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) {
            if (this.start == o.start) {
                return -1;
            } else {
                return this.start - o.start;
            }
        } else {
            return this.end - o.end;
        }
    }

}

• 여러 반례를 생각치 않아 시간이 많이 소요된 문제이다.
• 정렬을 생각한 것까지는 괜찮았으나, 뒤의 수만을 가지고 정렬을 진행하여 뒤의 수가 같은 경우의 처리가 되지 않았다.
• 정렬을 위해 TreeSet으로 진행하였으나, Set의 특성상 중복을 허용하지 않아 테스트케이스에서 계속 오류가 났었다. 같은 경우 그냥 -1을 리턴하는 것으로 처리하였다.