93402024-02-20 16:43:05ananászÓvodacpp17Wrong answer 28/50115ms26156 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>kimaradt_szerep;
	int cv;
	for (int i = 0; i < K; i++)
	{
		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;
	string sep = "";
	for (int i = 0; i < N; i++)
	{
		if (gyerekek[i].kapott == -1)
		{
			cout << sep << kimaradt_szerep[cv];
			gyerekek[i].kapott = kimaradt_szerep[cv];
			cv++;
		}
		else
		{
			cout << sep << gyerekek[i].kapott;
		}
		sep = " ";
	}
	cout << endl;
}
/*
struct szerep {
	int akt_db;
	int max_db;

};
struct gyerek {
	int kivant;
	int siras;
	int kapott;
	int id;
};
bool rendez(gyerek gy1, gyerek gy2) {
	return gy1.siras > gy2.siras;
}
bool rendez2(gyerek gy1, gyerek gy2) {
	return gy1.id < gy2.id;
}
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);
	for (int i = 0; i < N; i++)
	{
		cin >> x;
		gyerekek[i].kivant = x;
		gyerekek[i].kapott = -1;
		gyerekek[i].id = i;
	}
	for (int i = 0; i < N; i++)
	{
		cin >> x;
		gyerekek[i].siras = x;
	}
	sort(gyerekek.begin(), gyerekek.end(), rendez);
	int siras = 0;
	vector<int>kimaradt;
	for (int i = 0; i < N; i++)
	{
		if (szerepek[gyerekek[i].kivant - 1].akt_db < szerepek[gyerekek[i].kivant - 1].max_db)
		{
			szerepek[gyerekek[i].kivant - 1].akt_db++;
			gyerekek[i].kapott = gyerekek[i].kivant;
		}
		else
		{
			siras += gyerekek[i].siras;
			kimaradt.push_back(gyerekek[i].id);
		}
	}
	cout << siras << endl;
	int id = 0;
	string sep = "";
	sort(gyerekek.begin(), gyerekek.end(), rendez2);
	for (int i = 0; i < K; i++)
	{
		while (id<kimaradt.size()&&szerepek[i].akt_db < szerepek[i].max_db)
		{
			gyerekek[kimaradt[id]].kapott = i + 1;
			id++;
			szerepek[i].akt_db++;

		}

	}
	for (int i = 0; i < N; i++)
	{
		cout << sep << gyerekek[i].kapott;
		sep = " ";
	}
	cout << endl;
	/*for (int i = 0; i < N; i++)
	{
		if (gyerekek[i].kapott==-1)
		{
			while (szerepek[id].max_db == szerepek[id].akt_db)
			{
				id++;
			}
			//gyerekek[i].kapott=id+1;
			szerepek[id].akt_db++;
			cout <<sep<< id + 1;
		}
		else
		{
			cout << sep << gyerekek[i].kapott;
		}
		sep = " ";
	}
}*/

// 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
base28/50
1Accepted0/03ms1808 KiB
2Wrong answer0/07ms2568 KiB
3Accepted2/23ms2216 KiB
4Partially correct1/23ms2436 KiB
5Accepted2/23ms2636 KiB
6Accepted2/22ms2720 KiB
7Wrong answer0/23ms2752 KiB
8Partially correct1/23ms2952 KiB
9Accepted2/23ms3032 KiB
10Accepted2/23ms3164 KiB
11Accepted2/23ms3284 KiB
12Partially correct1/23ms3248 KiB
13Wrong answer0/23ms3564 KiB
14Accepted3/33ms3492 KiB
15Wrong answer0/314ms5052 KiB
16Wrong answer0/328ms7204 KiB
17Accepted3/339ms7648 KiB
18Wrong answer0/372ms15204 KiB
19Accepted3/370ms9904 KiB
20Accepted3/375ms10348 KiB
21Partially correct1/390ms11912 KiB
22Wrong answer0/4115ms26156 KiB