141522025-01-10 07:23:25NeMMaximális szorzat (50 pont)cpp17Wrong answer 42/5059ms1076 KiB
#include <bits/stdc++.h>
using namespace std;

int main() {
	int n, k, b, ln = 0;
	cin >> n >> k >> b;
	vector <int> a (n, INT_MIN);
	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}
	sort(a.begin(), a.end());
	a.push_back(INT_MAX);
	for (int i = 0; i < n+1; i++) {
		if (a[i] >= 0) {
			ln = i-1;
			break;
		}
	}
	

	if (ln < b-1) {
		cout << -1;
		return 0;
	}

	while (ln > b-1) {
		if (k == 0) {
			cout << -1;
			return 0;
		}

		if (a[ln] < 0) {
			a[ln]++;
			k--;
		} else {
			ln--;
		}
	}

	if (ln%2 == 0) {
		if ((a[ln]*-1) >= k) {
			k += a[ln];
			a[ln] = 0;
			ln--;
		} else {
			a[0] += k;
			k = 0;
		}
	}

	long long s = k;
	for (int i = ln+2; i < n+1; i++) {
		s += a[i-1];

		if (a[i] >= s/(i-(ln+1))) {
			int d = s/(i-(ln+1)), r = s%(i-(ln+1));
			for (int j = ln+1; j < i; j++) {
				k -= (d-a[j]);
				a[j] += (d-a[j]);

				if (r > 0) {
					a[j]++;
					r--;
				}
			}

			break;
		}
	}


	long long c = 1;
	for (int i = 0; i < n; i++) {
		c *= a[i];
		c = c % 1000000007;
	}
	cout << c;
	return 0;
}
SubtaskSumTestVerdictTimeMemory
base42/50
1Accepted0/01ms500 KiB
2Accepted0/01ms316 KiB
3Accepted0/01ms508 KiB
4Accepted0/01ms316 KiB
5Accepted0/04ms316 KiB
6Accepted2/21ms316 KiB
7Accepted2/21ms316 KiB
8Accepted2/21ms316 KiB
9Accepted2/21ms388 KiB
10Accepted2/26ms540 KiB
11Accepted2/252ms1052 KiB
12Accepted1/157ms1056 KiB
13Accepted1/11ms316 KiB
14Accepted1/14ms440 KiB
15Accepted1/120ms608 KiB
16Accepted1/121ms604 KiB
17Accepted1/120ms564 KiB
18Wrong answer0/110ms760 KiB
19Wrong answer0/141ms1072 KiB
20Wrong answer0/130ms1072 KiB
21Wrong answer0/159ms1060 KiB
22Wrong answer0/125ms1076 KiB
23Accepted1/159ms1056 KiB
24Accepted1/157ms1056 KiB
25Accepted2/21ms556 KiB
26Accepted2/24ms316 KiB
27Wrong answer0/229ms600 KiB
28Wrong answer0/129ms612 KiB
29Accepted2/223ms756 KiB
30Accepted1/159ms1076 KiB
31Accepted1/143ms1060 KiB
32Accepted2/21ms316 KiB
33Accepted2/259ms1052 KiB
34Accepted1/159ms1076 KiB
35Accepted2/259ms1056 KiB
36Accepted2/259ms1056 KiB
37Accepted2/259ms1072 KiB
38Accepted2/259ms1076 KiB
39Accepted1/11ms316 KiB