93422024-02-20 16:55:26ananászÓvodacpp17Wrong answer 33/50114ms21844 KiB
// ovoda.cpp : This file contains the 'main' function. Program execution begins and ends there.
//

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

struct gy {
	int id;
	int siras;
};
struct szerep {
	int akt_db;
	int max_db;
	vector<gy>gyerek;
};

struct gyerek {
	int kivant;
	int siras;
	int kapott;
	int id;
};
bool rendez(gy gy1, gy gy2) {
	return gy1.siras > gy2.siras;
}
int main()
{
	int N, K;
	cin >> N >> K;
	vector<szerep>szerepek(K);
	int x;
	for (int i = 0; i < K; i++)
	{
		cin >> x;
		szerepek[i].max_db = x;
		szerepek[i].akt_db = 0;
	}
	vector<gyerek>gyerekek(N);
	gy y;
	for (int i = 0; i < N; i++)
	{
		cin >> x;
		gyerekek[i].kivant = x;
		gyerekek[i].kapott = -1;
		gyerekek[i].id = i;
	}
	int siras = 0;
	for (int i = 0; i < N; i++)
	{
		cin >> x;
		gyerekek[i].siras = x;
		y.id = i;
		y.siras = x;
		szerepek[gyerekek[i].kivant - 1].gyerek.push_back(y);
		siras += x;
	}
	vector<int>nemvalasztott;
	vector<int>kimaradt_szerep;
	int cv;
	for (int i = 0; i < K; i++)
	{
		if (szerepek[i].gyerek.size() == 0)
		{
			nemvalasztott.push_back(i + 1);
			for (int j = 1; j < szerepek[i].max_db; j++)
			{
				kimaradt_szerep.push_back(i + 1);
			}
		}
		else
		{
			cv = 0;
			sort(szerepek[i].gyerek.begin(), szerepek[i].gyerek.end(), rendez);
			while (cv < szerepek[i].gyerek.size() && szerepek[i].akt_db < szerepek[i].max_db)
			{
				gyerekek[szerepek[i].gyerek[cv].id].kapott = i + 1;
				siras -= szerepek[i].gyerek[cv].siras;
				cv++;
				szerepek[i].akt_db++;

			}
			for (int j = szerepek[i].gyerek.size(); j < szerepek[i].max_db; j++)
			{
				kimaradt_szerep.push_back(i + 1);
			}
		}

	}
	cout << siras << endl;
	cv = 0;
	int cv2 = 0;
	string sep = "";
	for (int i = 0; i < N; i++)
	{
		if (gyerekek[i].kapott == -1)
		{
			if (cv < nemvalasztott.size())
			{
				cout << sep << nemvalasztott[cv];
				gyerekek[i].kapott = nemvalasztott[cv];
				cv++;
			}
			else
			{
				cout << sep << kimaradt_szerep[cv2];
				gyerekek[i].kapott = kimaradt_szerep[cv2];
				cv2++;
			}

		}
		else
		{
			cout << sep << gyerekek[i].kapott;
		}
		sep = " ";
	}
	cout << endl;
}


// Run program: Ctrl + F5 or Debug > Start Without Debugging menu
// Debug program: F5 or Debug > Start Debugging menu

// Tips for Getting Started: 
//   1. Use the Solution Explorer window to add/manage files
//   2. Use the Team Explorer window to connect to source control
//   3. Use the Output window to see build output and other messages
//   4. Use the Error List window to view errors
//   5. Go to Project > Add New Item to create new code files, or Project > Add Existing Item to add existing code files to the project
//   6. In the future, to open this project again, go to File > Open > Project and select the .sln file
SubtaskSumTestVerdictTimeMemory
base33/50
1Accepted0/03ms1684 KiB
2Accepted0/07ms2356 KiB
3Accepted2/23ms2060 KiB
4Accepted2/23ms2272 KiB
5Accepted2/22ms2392 KiB
6Accepted2/23ms2596 KiB
7Wrong answer0/22ms2568 KiB
8Accepted2/22ms2572 KiB
9Accepted2/23ms2696 KiB
10Accepted2/22ms2812 KiB
11Accepted2/23ms2904 KiB
12Accepted2/23ms3016 KiB
13Wrong answer0/23ms3048 KiB
14Accepted3/33ms3024 KiB
15Wrong answer0/314ms4212 KiB
16Wrong answer0/328ms6444 KiB
17Accepted3/339ms7280 KiB
18Wrong answer0/365ms14560 KiB
19Accepted3/368ms9368 KiB
20Accepted3/374ms9748 KiB
21Accepted3/390ms11300 KiB
22Wrong answer0/4114ms21844 KiB