제출 #591975

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
591975		starchan	Garage (IOI09_garage)	C++17	100 / 100	1 ms	340 KiB

garage

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define in pair<int, int>
#define f first
#define s second
#define pb push_back
#define pob pop_back
#define INF (int)1e17
#define MX (int)3e5+5
#define fast() ios_base::sync_with_stdio(false); cin.tie(NULL)
signed main()
{
	fast();
	int n, m;
	cin >> n >> m;
	vector<int> r(n+1);
	set<int> empty;
	for(int i = 1; i <= n; i++)
	{
		cin >> r[i];
		empty.insert(i);
	}
	vector<int> which(m+1);
	vector<int> w(m+1);
	for(int i = 1; i <= m; i++)
		cin >> w[i];
	int t = 2*m;
	int ans = 0;
	queue<int> welp;
	while(t--)
	{
		int x;
		cin >> x;
		int u = abs(x);
		if(x > 0)
		{
			if(empty.empty())
			{
				welp.push(x);
				continue;
			}
			int k = *empty.begin();
			which[u] = k;
			empty.erase(empty.begin());
			ans+=(w[u]*r[k]);
		}
		else
		{
			empty.insert(which[u]);
			while(welp.size() && empty.size())
			{
				int u = welp.front();
				welp.pop();
				int k = *empty.begin();
				which[u] = k;
				empty.erase(empty.begin());
				ans+=(w[u]*r[k]);
			}
		}
	}
	cout << ans;
	return 0;
}

• Subtask 1: 100 / 100