244092026-02-11 09:41:56SzeredaiNatanaelÓvodacpp17Hibás válasz 26/50500ms3260 KiB
#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main()
{
	int k, n; cin >> k >> n;
	vector<pair<int, bool>> maxi(n + 1, { 0, false });
	for (int i = 1; i <= n; ++i) {
		cin >> maxi[i].first;
	}
	vector<int> szerep(k + 1);
	vector<int> ans(k + 1, 0);
	for (int i = 1; i <= k; ++i) {
		cin >> szerep[i];
	}
	vector<pair<int, int>> siras(k + 1);
	for (int i = 1; i <= k; ++i) {
		cin >> siras[i].first;
		siras[i].second = i;
	}

	int t = 0;
	sort(siras.begin(), siras.end());
	for (int i = k; i >= 1; --i) {
		int j = szerep[siras[i].second];
		if (maxi[j].first > 0) {
			--maxi[j].first;
			maxi[j].second = true;
			ans[siras[i].second] = j;
		}
		else t += siras[i].first;
	}

	int i = 1;
	for (auto& e : ans) {
		if (e == 0) {
			for (int j = i; j <= n; ++j){
				if (maxi[j].first > 0 && !maxi[j].second) {
					--maxi[j].first;
					maxi[j].second = true;
					e = j;

					if (maxi[j].first == 0) {
						i = j + 1;
					}
					break;
				}
			}
		}
	}

	bool nincs = false, van = true;
	for (auto& e : ans) {
		if (e == 0) {
			nincs = true;
		}
	}
	for (auto& e : maxi) {
		if (!e.second) {
			van = false;
		}
	}

	if (nincs) {
		i = 1;
		for (auto& e : ans) {
			if (e == 0) {
				for (int j = i; j <= n; ++j) {
					if (maxi[j].first > 0) {
						--maxi[j].first;
						e = j;

						if (maxi[j].first == 0) {
							i = j + 1;
						}
						break;
					}
				}
			}
		}
	}

	cout << t << "\n";
	for (int i = 1; i <= k; ++i) {
		cout << ans[i] << " ";
	}
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base26/50
1Elfogadva0/01ms316 KiB
2Hibás válasz0/04ms316 KiB
3Elfogadva2/21ms316 KiB
4Részben helyes1/21ms316 KiB
5Részben helyes1/21ms316 KiB
6Elfogadva2/21ms360 KiB
7Hibás válasz0/21ms508 KiB
8Elfogadva2/21ms328 KiB
9Elfogadva2/21ms316 KiB
10Elfogadva2/21ms316 KiB
11Részben helyes1/21ms316 KiB
12Részben helyes1/21ms316 KiB
13Hibás válasz0/22ms316 KiB
14Elfogadva3/32ms500 KiB
15Hibás válasz0/330ms640 KiB
16Hibás válasz0/326ms1000 KiB
17Elfogadva3/337ms1164 KiB
18Hibás válasz0/359ms2028 KiB
19Elfogadva3/3160ms2100 KiB
20Elfogadva3/3123ms2208 KiB
21Időlimit túllépés0/3500ms2084 KiB
22Hibás válasz0/4104ms3260 KiB