백준 5464번 주차장 C++

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

5464번: 주차장

 

들어오는 차일 경우 : 빈자리가 있으면 그곳에 넣고, 없으면 대기 큐에 넣는다.

나가는 차일 경우 : 나가면서 총 가격을 계산해서 더하고, 만약 대기 큐가 비어있지 않다면 하나를 꺼내서 방금 나간 곳에 채워넣는다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int n, m;

vector<int> price;
vector<int> carWeight;
vector<int> currParkingLot;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> n >> m;
	price = vector<int>(n);
	currParkingLot = vector<int>(n, -1);
	carWeight = vector<int>(m);
	for (int i = 0; i < n; i++) cin >> price[i];
	for (int i = 0; i < m; i++) cin >> carWeight[i];
	int64 sum = 0;
	queue<int> waitQueue;
	for (int i = 0; i < m * 2; i++) {
		int carIdx;
		bool out = false;
		cin >> carIdx;
		if(carIdx < 0) {
			carIdx = -carIdx;
			out = true;
		}
		carIdx--;

		if (out) {
			int parkingIdx;
			for (int i = 0; i < n; i++) {
				if (currParkingLot[i] == carIdx) {
					parkingIdx = i;
					break;
				}
			}
			currParkingLot[parkingIdx] = -1;
			sum += carWeight[carIdx] * price[parkingIdx];
			if (!waitQueue.empty()) {
				int nextCar = waitQueue.front();
				waitQueue.pop();
				currParkingLot[parkingIdx] = nextCar;
			}
		}
		else {
			int emptyPlace = -1;
			for (int i = 0; i < n; i++) {
				if (currParkingLot[i] == -1) {
					emptyPlace = i;
					break;
				}
			}
			if (emptyPlace == -1) {
				waitQueue.push(carIdx);
			}
			else {
				currParkingLot[emptyPlace] = carIdx;
			}
		}
	}
	cout << sum;
};