Submission #885876

#	Time	Username	Problem	Language	Result	Execution time	Memory
885876		Matjaz	Garage (IOI09_garage)	C++14	100 / 100	2 ms	348 KiB

garage

//
//  GarageIOI2009.cpp
//  
//
//  Created by Matjaz Leonardis on 10/12/2023.
//

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

int N,M;

int main(){
    
    cin >> N >> M;
    
    vector<int> R(N);
    vector<int> W(M);
    
    for (int i=0;i<N;i++) cin >> R[i];
    for (int i=0;i<M;i++) cin >> W[i];
    
    priority_queue<int> free_spaces;
    queue<int> waiting;
    for (int i=0;i<N;i++) free_spaces.push(-i);
    
    vector<int> place(M);
    
    for (int i=0;i<2*M;i++){
        int event;
        cin >> event;
        if (event < 0){
            event *= -1;
            event--;
            free_spaces.push(-place[event]);
            if (!waiting.empty()){
                place[waiting.front()] = - free_spaces.top();
                waiting.pop();
                free_spaces.pop();
            }
        } else {
            event--;
            if (free_spaces.empty()){
                waiting.push(event);
            } else {
                place[event] = - free_spaces.top();
                free_spaces.pop();
            }
        }
    }
    
    int total = 0;
    for (int i=0;i<M;i++) total += W[i] * R[place[i]];
    
    cout << total << endl;
    
    
    return 0;
}

• Subtask 1: 100 / 100