201192026-01-02 11:21:18PappMatyasÓvodacpp17Wrong answer 12/50499ms2540 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)
				{
					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
base12/50
1Accepted0/01ms316 KiB
2Wrong answer0/04ms316 KiB
3Accepted2/21ms316 KiB
4Partially correct1/21ms500 KiB
5Wrong answer0/21ms508 KiB
6Accepted2/21ms316 KiB
7Wrong answer0/21ms316 KiB
8Wrong answer0/21ms316 KiB
9Accepted2/21ms316 KiB
10Accepted2/21ms316 KiB
11Wrong answer0/21ms316 KiB
12Wrong answer0/21ms316 KiB
13Wrong answer0/22ms316 KiB
14Wrong answer0/32ms316 KiB
15Wrong answer0/327ms660 KiB
16Wrong answer0/3263ms880 KiB
17Accepted3/332ms1028 KiB
18Wrong answer0/356ms1716 KiB
19Time limit exceeded0/3485ms1484 KiB
20Time limit exceeded0/3476ms1452 KiB
21Time limit exceeded0/3499ms1712 KiB
22Wrong answer0/493ms2540 KiB