알고리즘

백준 19621번 회의실 배정 2 (JAVA)

박카스마시며코딩 2023. 12. 31. 15:56

https://www.acmicpc.net/problem/19621

 

19621번: 회의실 배정 2

서준이는 아빠로부터 N개의 회의와 하나의 회의실을 선물로 받았다. 각 회의는 시작 시간, 끝나는 시간, 회의 인원이 주어지고 한 회의실에서 동시에 두 개 이상의 회의가 진행될 수 없다. 단,

www.acmicpc.net

 

저는 DP를 이용해 문제를 풀었습니다.

DP를 통해 이전에 계산한 것을 다시 계산하지 않도록 하여 문제를 풀었습니다.

 

package BOJ.dp;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.*;

public class BOJ_19621_2 {
    private static class Meeting{
        int startTime;
        int endTime;
        int peopleCnt;
        public Meeting(int startTime,int endTime, int peopleCnt){
            this.startTime = startTime;
            this.endTime = endTime;
            this.peopleCnt = peopleCnt;
        }
    }
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        int[][] time = new int[n][3];
        List<Meeting> meetings = new ArrayList<>();
        for(int i = 0 ; i < n ; i++){
            StringTokenizer st = new StringTokenizer(br.readLine());
            int start = Integer.parseInt(st.nextToken());
            int end = Integer.parseInt(st.nextToken());
            int peopleCnt = Integer.parseInt(st.nextToken());
            meetings.add(new Meeting(start,end,peopleCnt));
        }
        meetings.sort((v1,v2)->{
            return v1.startTime - v2.startTime;
        });
        int[] dp = new int[n];
        Arrays.fill(dp,NOT_VALID);
        int result = cal(0,meetings,n,dp);
        System.out.println(result);
    }
    private static final int NOT_VALID = -1;
    private static int cal(int depth, List<Meeting> meetings, int n,int[] dp) {
        if(depth >= n){
            return 0;
        }
        if(dp[depth] != NOT_VALID){
            return dp[depth];
        }
        int result = 0;
        for(int i = depth ; i < n ; i++){
            if(meetings.get(depth).endTime <= meetings.get(i).startTime){
                result = Math.max(result,cal(i,meetings,n,dp) + meetings.get(depth).peopleCnt);
            }
        }
        result = Math.max(result,cal(n,meetings,n,dp) + meetings.get(depth).peopleCnt);
        result = Math.max(result,cal(depth+1,meetings,n,dp));
        dp[depth] = result;
        return result;
    }
}