PAT 甲级 1026 JAVA

PAT 甲级 1026 JAVA

import java.util.*; public class Main { static Comparator2 com = new Comparator2(); static List<Table> total_tables = new ArrayList<>(); static List<Table> vip_tables = new ArrayList<>(); static Queue<Player> normal_queue = new LinkedList<>(); static Queue<Player> vip_queue = new LinkedList<>(); static List<Player> total_member = new ArrayList<>(); public static void main(String[] args) { Scanner sc = new Scanner(System.in); int N = sc.nextInt(); for (int i = 0; i < N; i++) { Player p = new Player(); String arrTime = sc.next(); String[] spl = arrTime.split(":"); int hour = Integer.parseInt(spl[0]); int min = Integer.parseInt(spl[1]); int sec = Integer.parseInt(spl[2]); p.arrivingTime = hour * 3600 + min * 60 + sec; p.playTime = sc.nextInt() * 60; if (p.playTime > 120 * 60) { p.playTime = 120 * 60; } p.VIP = sc.nextInt(); total_member.add(p); } int K = sc.nextInt(); for (int i = 0; i < K; i++) { total_tables.add(new Table(i)); } int M = sc.nextInt(); for (int i = 0; i < M; i++) { vip_tables.add(total_tables.get(sc.nextInt() - 1)); } for (int i = 8 * 3600 - 1; i < 21 * 3600; i++) { process(i, total_member); } Collections.sort(total_member, com); for (Player p : total_member) { if (p.serveTime == 0) { continue; } System.out.println(exchangeTimeFormat(p.arrivingTime) + " " + exchangeTimeFormat(p.serveTime) + " " + Math.round((double)p.waitTime / 60)); } StringBuffer strb = new StringBuffer(); for (Table t : total_tables) { strb.append(t.times); strb.append(" "); } System.out.println(strb.deleteCharAt(strb.length() - 1).toString()); } private static String exchangeTimeFormat(int time) { StringBuffer strb = new StringBuffer(); int hour = time / 3600; int min = (time % 3600) / 60; int sec = time % 3600 % 60; if (hour < 10) { strb.append(0); } strb.append(hour); strb.append(":"); if (min < 10) { strb.append(0); } strb.append(min); strb.append(":"); if (sec < 10) { strb.append(0); } strb.append(sec); return strb.toString(); } private static void process(int current_time, List<Player> total_member) { for (Player p : total_member) { if (p.endTime == current_time) { total_tables.get(p.table_number).isUsed = 0; } if (p.arrivingTime == current_time) { if (p.VIP == 1) { vip_queue.add(p); } else { normal_queue.add(p); } } } Player cur; if (!vip_queue.isEmpty()) { cur = vip_queue.peek(); for (Table t : vip_tables) { if (t.isUsed == 0) { t.isUsed = 1; t.times++; cur.serveTime = current_time; cur.table_number = t.ID; cur.waitTime = current_time - cur.arrivingTime; cur.endTime = cur.serveTime + cur.playTime; if (cur.endTime > 21 * 3600) cur.endTime = 21 * 3600; vip_queue.remove(); return; } } for (Table t : total_tables) { if (t.isUsed == 0) { t.isUsed = 1; t.times++; cur.serveTime = current_time; cur.table_number = t.ID; cur.waitTime = current_time - cur.arrivingTime; cur.endTime = cur.serveTime + cur.playTime; if (cur.endTime > 21 * 3600) cur.endTime = 21 * 3600; vip_queue.remove(); return; } } } else if (!normal_queue.isEmpty()) { cur = normal_queue.peek(); for (Table t : total_tables) { if (t.isUsed == 0) { t.isUsed = 1; t.times++; cur.serveTime = current_time; cur.table_number = t.ID; cur.waitTime = current_time - cur.arrivingTime; cur.endTime = cur.serveTime + cur.playTime; if (cur.endTime > 21 * 3600) cur.endTime = 21 * 3600; normal_queue.remove(); return; } } } else { return; } } private static class Player { int arrivingTime; int serveTime; int playTime; int waitTime; int endTime; int VIP; int table_number; } private static class Table { int isUsed; int times; int ID; Table(int ID) { this.ID = ID; } } private void decide (int currentTime, Player p1) { if (p1.VIP == 1) { } } private static class Comparator2 implements Comparator<Player> { @Override public int compare(Player p1, Player p2) { if (p1.serveTime > p2.serveTime) { return 1; } else { return -1; } } } }