[백준] 1700번 멀티탭 스케줄링 Java

문제 링크

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

알고리즘: 그리디

문제

기숙사에서 살고 있는 준규는 한 개의 멀티탭을 이용하고 있다. 준규는 키보드, 헤어드라이기, 핸드폰 충전기, 디지털 카메라 충전기 등 여러 개의 전기용품을 사용하면서 어쩔 수 없이 각종 전기용품의 플러그를 뺐다 꽂았다 하는 불편함을 겪고 있다. 그래서 준규는 자신의 생활 패턴을 분석하여, 자기가 사용하고 있는 전기용품의 사용순서를 알아내었고, 이를 기반으로 플러그를 빼는 횟수를 최소화하는 방법을 고안하여 보다 쾌적한 생활환경을 만들려고 한다.

예를 들어 3 구(구멍이 세 개 달린) 멀티탭을 쓸 때, 전기용품의 사용 순서가 아래와 같이 주어진다면,

1.키보드
2.헤어드라이기
3.핸드폰 충전기
4.디지털 카메라 충전기
5.키보드
6.헤어드라이기

키보드, 헤어드라이기, 핸드폰 충전기의 플러그를 순서대로 멀티탭에 꽂은 다음 디지털 카메라 충전기 플러그를 꽂기 전에 핸드폰충전기 플러그를 빼는 것이 최적일 것이므로 플러그는 한 번만 빼면 된다.

풀이

먼저 멀티탭이 비어있다면, 스케줄링에 맞게 채워준다.
멀티탭을 다 채웠다면, 다음 스케줄부터 어떤 플로그를 뽑아야 할지 선택해야 하는데
플로그를 뽑는 우선순위를 보면

1.다음 스케줄이 없는 전기용품
2.스케줄이 제일 뒤에 위치한 전기용품 => 가장 오랫동안 사용하지 않는 전기용품
(교체 알고리즘 OPT)
이 우선순위로 풀면 최소 횟수를 만들 수 있다.

소스 코드

import java.io.*;
import java.util.*;

class Multitap {
    int v,ns;
    Multitap(int v, int ns) {
        this.v = v;
        this.ns = ns;
    }
}

public class Main {
    static Multitap multitap[];
    static int schedule[];
    static ArrayList<Integer> ea_schedule[];
    static int N,K;
    static int ans = 0;
    static int socket = 0;
    public static void main(String args[]) throws IOException {
      BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
      StringTokenizer st = new StringTokenizer(br.readLine());
      N = Integer.parseInt(st.nextToken());
      K = Integer.parseInt(st.nextToken());
      multitap = new Multitap[N];
      schedule = new int[K];
      ea_schedule = new ArrayList[K+1];
      for(int i=0; i<K+1; i++) {
          ea_schedule[i] = new ArrayList<Integer>();
      }
      StringTokenizer sti = new StringTokenizer(br.readLine());
      for(int i=0; i<K; i++) {
          int n = Integer.parseInt(sti.nextToken());
          schedule[i] = n;
          ea_schedule[n].add(i);
      }
      
      for(int i=0; i<K; i++) {
          int ea = schedule[i];
          int ns = next_schedule(ea, i);
          int mi = check(ea);
          if(socket < N) {
              if(mi != -1) {
                  //이미 꽂혀있음.
                  multitap[mi] = new Multitap(ea, ns);
              } else {
                  multitap[socket] = new Multitap(ea, ns);
                  socket += 1;
              }
          } else {
              if(mi != -1) {
                  //이미 꽂혀있음.
                  multitap[mi] = new Multitap(ea, ns);
              } else {
                  int emi = s_compare();
                  multitap[emi] = new Multitap(ea, ns);
                  ans += 1;
              }
          }
      }
      System.out.println(ans);
    }
    
    static int check(int n) {
        for(int i=0; i<N; i++) {
            if(multitap[i] != null) {
                if(multitap[i].v == n) {
                    return i;
                }
            }
        }
        return -1;
    }
    
    static int next_schedule(int n, int cs) {
        for(int i=0; i<ea_schedule[n].size(); i++) {
            if(ea_schedule[n].get(i) > cs) {
                return ea_schedule[n].get(i);
            }
        }
        return 101;
    }
    
    static int s_compare() {
        int max = multitap[0].ns; 
        int max_i = 0;
        for(int i=0; i<N; i++) {
           if(max < multitap[i].ns) {
               max = multitap[i].ns;
               max_i = i;
           }
        }
        return max_i;
    }
}