본문으로 바로가기

[알고리즘 문제]결혼식 - Greedy Algorithm

category Algorithm/Greedy 2021. 12. 6. 07:35
728x90
반응형

▷ 문제

현수는 다음 달에 결혼을 합니다.

현수는 결혼식 피로연을 장소를 빌려 3일간 쉬지 않고 하려고 합니다.

피로연에 참석하는 친구들 N명의 참석하는 시간정보를 현수는 친구들에게 미리 요구했습니다.

각 친구들은 자신이 몇 시에 도착해서 몇 시에 떠날 것인지 현수에게 알려주었습니다.

현수는 이 정보를 바탕으로 피로연 장소에 동시에 존재하는 최대 인원수를 구하여 그 인원을 수용할 수 있는 장소를 빌리려고 합니다. 여러분이 현수를 도와주세요.

만약 한 친구가 오는 시간 13, 가는시간 15라면 이 친구는 13시 정각에 피로연 장에 존재하는 것이고 15시 정각에는 존재하지 않는다고 가정합니다.

* 입력

첫째 줄에 피로연에 참석할 인원수 N(5<=N<=100,000)이 주어집니다.

두 번째 줄부터 N줄에 걸쳐 각 인원의 오는 시간과 가는 시간이 주어집니다.

시간은 첫날 0시를 0으로 해서 마지막날 밤 12시를 72로 하는 타임라인으로 오는 시간과 가는 시간이 음이 아닌 정수로 표현됩니다.

* 출력

첫째 줄에 피로연장에 동시에 존재하는 최대 인원을 출력하세요.

▷ 입력 예시

5

14 18

12 15

15 20

20 30

5 14

▷ 출력 예시

2

▷ 풀이

import java.util.*;

class Time implements Comparable<Time>{
    public int time;
    public int state;

    Time(int time, char state){
        this.time = time;
        this.state = state;
    }
    
    @Override
    public int compareTo(Time ob){
        if(this.time == ob.time){
          return this.state - ob.state; // this에서 매개변수를 빼면 오름차순
        } else{
          return this.time - ob.time; 
        }
    }
}

class Main{
    public int solution(ArrayList<Time> arr){
        int answer = Integer.MIN_VALUE;
        Collections.sort(arr);

        int cnt = 0;
        for(Time ob : arr){
            if(ob.state == 's'){
              cnt++;
            } else{
              cnt--;
            }
            answer = Math.max(answer, cnt);
        }

        return answer;
    }

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

        int n = kb.nextInt();
        ArrayList<Time> arr = new ArrayList<>();

        for(int i = 0; i < n; i++){
            int sT = kb.nextInt();
            int eT = kb.nextInt();
            arr.add(new Time(sT, 's'));
            arr.add(new Time(eT, 'e'));
        }

        kb.close();
        System.out.println(main.solution(arr));
    }
}

▷ 메모

  • 시간이 같을 경우 'e'가 's'보다 먼저 나와야하므로 상태값을 오름차순한다.
반응형