201202026-01-02 11:35:09PappMatyasÓvodacpp17Wrong answer 24/50499ms2700 KiB
#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

vector<int> maxm;
vector<int> Tot;
vector<int> wants;
vector<int> losses;
vector<bool> req;

int main()
{
	int loss = 0;
	int n, k;
	cin >> n >> k;
	for (int i = 0; i < k; i++)
	{
		int l;
		cin >> l;
		maxm.push_back(l);
		Tot.push_back(0);
	}
	for (int i = 0; i < n; i++)
	{
		int a;
		cin >> a;
		a--;
		req.push_back(false);
		wants.push_back(a);
	}
	for (int i = 0; i < n; i++)
	{
		int b;
		cin >> b;
		losses.push_back(b);
	}
	for (int i = 0; i < n; i++)
	{
		int index = wants[i]; // 0 - (k - 1)
		if (Tot[index] == maxm[index])
		{
			int min = 10001;
			int s = -1;
			for (int j = 0; j <= i; j++)
			{
				int jndex = wants[j];
				if (jndex == index && !req[j])
				{
					if (losses[j] < min)
					{
						s = j;
						min = losses[j];
					}
				}
			}
			req[s] = true;
			loss += min;
		}
		else
		{
			Tot[index]++;
		}
	}
	cout << loss << endl;
	for (int i = 0; i < n; i++)
	{
		if (!req[i])
		{
			cout << wants[i] + 1 << " ";
		}
		else
		{
			for (int l = 0; l < k; l++)
			{
				if (Tot[l] < maxm[l])
				{
					cout << l + 1 << " ";
					Tot[l]++;
					break;
				}
			}
		}
	}
}
SubtaskSumTestVerdictTimeMemory
base24/50
1Accepted0/01ms508 KiB
2Wrong answer0/04ms316 KiB
3Accepted2/21ms316 KiB
4Partially correct1/21ms508 KiB
5Accepted2/21ms316 KiB
6Accepted2/21ms316 KiB
7Wrong answer0/21ms316 KiB
8Partially correct1/21ms316 KiB
9Accepted2/21ms316 KiB
10Accepted2/21ms316 KiB
11Accepted2/21ms508 KiB
12Partially correct1/21ms400 KiB
13Wrong answer0/21ms316 KiB
14Accepted3/32ms316 KiB
15Wrong answer0/319ms664 KiB
16Wrong answer0/3182ms1076 KiB
17Accepted3/332ms1148 KiB
18Wrong answer0/354ms1652 KiB
19Accepted3/3296ms1488 KiB
20Time limit exceeded0/3479ms1448 KiB
21Time limit exceeded0/3499ms1532 KiB
22Wrong answer0/494ms2700 KiB