[JAVA] SWEA D3 / 9280. 진용이네 주차타워

SW Expert Academy

 

코드

• getParkingSpot: 주차할 수 있는 공간 반환
• getParkedSpot: 특정 차량이 주차된 공간 반환

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

class Solution {

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int T = Integer.parseInt(br.readLine());
        for (int t = 1; t <= T; t++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int n = Integer.parseInt(st.nextToken()); // 차량 개수
            int m = Integer.parseInt(st.nextToken()); // 주차장 개수

            int[] seatPrices = new int[n];
            for (int i = 0; i < n; i++) {
                seatPrices[i] = Integer.parseInt(br.readLine());
            }

            int[] carPrices = new int[m];
            for (int i = 0; i < m; i++) {
                carPrices[i] = Integer.parseInt(br.readLine());
            }

            int[] seatState = new int[n]; // 주차장 현황
            Arrays.fill(seatState, -1);
            Queue<Integer> waitCars = new LinkedList<>(); // 대기 중인 차량

            int sum = 0;
            for (int i = 0; i < 2 * m; i++) {
                int carNum = Integer.parseInt(br.readLine()) - 1;
                if (carNum >= 0) { // 주차장 들어올 때
                    if (isNotParkingFull(seatState)) { // 주차할 자리가 있을 때
                        int seatNumber = getParkingSpot(seatState);
                        seatState[seatNumber] = carNum;
                    } else { // 주차할 자리가 없을 때
                        waitCars.offer(carNum);
                    }
                } else { // 주차장 나갈 때
                    carNum = Math.abs(carNum) - 2;
                    int parkedSpot = getParkedSpot(seatState, carNum);
                    sum += seatPrices[parkedSpot] * carPrices[carNum]; // 가격 정산
                    seatState[parkedSpot] = -1; // 자리 비운다
                    if (!waitCars.isEmpty()) {
                        int pollCar = waitCars.poll();
                        seatState[parkedSpot] = pollCar;
                    }
                }
            }

            System.out.println("#" + t + " " + sum);
        }
    }

    static boolean isNotParkingFull(int[] seatState) {
        for (int i = 0; i < seatState.length; i++) {
            if (seatState[i] == -1) {
                return true;
            }
        }
        return false;
    }

    static int getParkingSpot(int[] seatState) {
        for (int i = 0; i < seatState.length; i++) {
            if (seatState[i] == -1) {
                return i;
            }
        }
        return -1;
    }

    static int getParkedSpot(int[] seatState, int carNum) {
        for (int i = 0; i < seatState.length; i++) {
            if (seatState[i] == carNum) {
                return i;
            }
        }
        return -1;
    }
}