제출 #569412

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
569412		Aurora2005	Garage (IOI09_garage)	C++14	100 / 100	2 ms	212 KiB

garage

#include<bits/stdc++.h>
using namespace std;

int main(){
  int N,M;
  cin >> N >> M;
  vector<long long int> R(N);
  for(int i=0;i<N;i++) cin >> R[i];
  vector<long long int> W(M);
  for(int i=0;i<M;i++) cin >> W[i];
  priority_queue<int,vector<int>,greater<int>> que1;
  for(int i=0;i<N;i++) que1.push(i);
  queue<int> que2;
  vector<int> P(M);
  long long int ans = 0;
  for(int i=0;i<2*M;i++){
    int C;
    cin >> C;
    if(C > 0){
      C--;
      if(que1.size() == 0) que2.push(C);
      else{
        int a = que1.top();
        que1.pop();
        ans  += R[a]*W[C];
        P[C] = a;
      }
    }
    else{
      C++;
      C *= -1;
      int a = P[C];
      que1.push(a);
      if(que2.size() > 0){
        int b = que1.top();
        int c = que2.front();
        ans += R[b]*W[c];
        P[c] = b;
        que1.pop();
        que2.pop();
      }
    }
  }
  cout << ans << endl;
}

• Subtask 1: 100 / 100