247622026-02-14 23:37:13ubormaciMaximális szorzat (50 pont)cpp17Accepted 50/50226ms11976 KiB
#pragma GCC optimize("O2")
// #pragma GCC optimize("O1","O2","O3","Ofast","unroll-loops")
// #pragma GCC target("sse","sse2","sse3","sse4.1","sse4.2","avx","avx2","fma")

#include <iostream>
#include <algorithm> // for sort, mainly
#include <vector>
#include <map>
#include <set>
#include <cmath>
#include <array>
#include <string>
#include <cstdio>
#include <iterator>
#include <unordered_set>
#include <cstdint> // for int64_t, int32_t, etc
#include <queue>
#include <stack>
#include <deque>
#include <numeric> // gcd, lcm
#include <fstream>
#include <bitset> // for bitset
#include <iomanip>
#include <cassert> // for set with custom ordering
#include <type_traits> // for set with custom ordering
#include <utility> // for swap, forward, etc
using namespace std;

template<typename A, typename B> ostream& operator<<(ostream &os, const pair<A, B> &p) { return os << '(' << p.first << ", " << p.second << ')'; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> ostream& operator<<(ostream &os, const T_container &v) { os << '{'; string sep; for (const T &x : v) os << sep << x, sep = ", "; return os << '}'; }
void dbg_out() { cout << endl; }
template<typename Head, typename... Tail> void dbg_out(Head H, Tail... T) { cout << ' ' << H; dbg_out(T...); }
#ifdef LOCAL
#define dbg(...) cout << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__)
#else
#define dbg(...)
#endif

/*

notes:

int64_t
stoi(string s) -> string to int
to_string() -> int (or else) to string

vector declaration:
vector<ll> v(n, 0)
vector<vector<ll>> v(n, vector<ll>(n, 0));

{if statement} ? {truth value} : {false value}

#ifdef LOCAL
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
#endif

std::lcm(ll a, ll b), std::gcd(int a, int b)

cout << fixed << setprecision(n);

set with custom ordering
set<ll, decltype(&cmp)> qu(cmp);

*/

typedef int64_t ll;

const ll md = 1e9 + 7;

void solve() {

	ll n, k, b;
	cin >> n >> k >> b;

	ll negc = 0;

	vector<ll> v(n, 0);
	for(ll i = 0; i < n; i++) {
		cin >> v[i];
		if(v[i] < 0) {
			negc++;
		}
	}

	if(negc < b) {
		cout << "-1";
		return;
	}

	if(b == n) {
		// egyedi eset
		// nincs pozitiv szam amit novelgessunk
		// minden novelessel veszitunk a szorzat ertekebol

		// ugyanazt csinaljuk, igazabol, csak forditva
		// az egeszet levisszuk (-1)-re, megnezzuk, hany opearacio volt
		// ha tobb mint k, szupi, visszaepitjuk oket
		// ha kevesebb, szivas, nem lehet
		
		ll down = 0;
		map<ll,ll> mp;
		set<ll> s;
		for(ll i = 0; i < n; i++) {
			down += abs(-1 - v[i]);
			mp[v[i]]++;
			s.insert(v[i]);
		}

		if(down < k) {
			cout << "-1";
			return;
		}
    }

	sort(v.begin(), v.end());
	for(ll i = n - 1; i >= 0; i--) {
		if(v[i] < 0 && negc > b) {
			
			ll nd = abs(v[i]);
			if(k >= nd) {
				v[i] = 0;
				negc--;
				k -= nd;
			}else{
				cout << "-1";
				return;
			}
		}
	}

	//cerr << "\nv=" << v;

	map<ll,ll> mp;
	set<ll> s;

	for(ll i = 0; i < n; i++) {
		if(v[i] >= 0 || b == n) {
			mp[v[i]]++;
			s.insert(v[i]);
		}
	}

    auto giveall = [&mp, &k](ll key) {

        ll val = mp[key];
        
        //cerr << "\nall are equal, final element";

        ll ext = k % val;
        ll all = k / val;

        //cerr << "\next=" << ext << "; all=" << all;

        if(all > 0) {
            mp[key + all] += val - ext;
            mp[key + all + 1] += ext;
            mp[key] = 0;
        }else{
            mp[key] -= ext;
            mp[key + 1] += ext;
        }

        k = 0;

    };

	for(auto & [key, val] : mp) {
		
		//cerr << "\nkey=" << key << "; val=" << val << "; s=" << s;

		if(val == 0) {
			continue;
		}

		if(k == 0) {
			break;
		}

		if(s.count(key) == false) {
			continue;
		}

		// tuddjuk-e novelni a kovetkezoig
		// ha nem, egyenkent mennyire tuddjuk novelni oket?
		
		if(key == *s.rbegin()) {

			giveall(key);
			break;
		}

		ll nxt = *s.upper_bound(key);
		
		ll dis = abs(nxt - key);

		if(k >= dis * val) {
			k -= dis * val;
			mp[nxt] += val;
			mp[key] = 0;
			s.erase(key);
		}else{
			//s.erase(key);
			giveall(key);
			break;
		}
	}

	//cerr << "\nmp=" << mp;

	ll j = n-1;
	for(auto it = mp.rbegin(); it != mp.rend(); it++) {

		auto h = *it;
		//cerr << "\nh=" << h;
		for(ll i = 0; i < h.second; i++) {
			v[j] = h.first;
			j--;
		}
	}

	//cerr << "\nv=" << v;

	ll ans = 1;

	for(ll i = 0; i < n; i++) {

		if(v[i] < 0) {
			ans = (ans * ((v[i] * v[i+1]) % md)) % md;
			i++;
		}else if(v[i] >= 0) {
		    v[i] %= md;
			ans = (ans * v[i]) % md;
		}
	}

	cout << ans;

}

int main()
{
	std::ios_base::sync_with_stdio(false);
	//cin.tie(nullptr);
	//cout.tie(nullptr);

	solve();

	return 0;
}
SubtaskSumTestVerdictTimeMemory
base50/50
1Accepted0/01ms508 KiB
2Accepted0/01ms316 KiB
3Accepted0/01ms316 KiB
4Accepted0/01ms316 KiB
5Accepted0/03ms316 KiB
6Accepted2/21ms316 KiB
7Accepted2/21ms372 KiB
8Accepted2/21ms316 KiB
9Accepted2/21ms316 KiB
10Accepted2/27ms1360 KiB
11Accepted2/293ms11976 KiB
12Accepted1/1101ms11968 KiB
13Accepted1/11ms316 KiB
14Accepted1/14ms820 KiB
15Accepted1/112ms732 KiB
16Accepted1/117ms1588 KiB
17Accepted1/110ms820 KiB
18Accepted1/16ms820 KiB
19Accepted1/170ms2100 KiB
20Accepted1/128ms1224 KiB
21Accepted1/1226ms11928 KiB
22Accepted1/113ms1076 KiB
23Accepted1/159ms6596 KiB
24Accepted1/193ms10976 KiB
25Accepted2/21ms316 KiB
26Accepted2/23ms316 KiB
27Accepted2/278ms6144 KiB
28Accepted1/187ms6212 KiB
29Accepted2/28ms820 KiB
30Accepted1/128ms1076 KiB
31Accepted1/123ms1076 KiB
32Accepted2/21ms316 KiB
33Accepted2/228ms1076 KiB
34Accepted1/128ms976 KiB
35Accepted2/228ms1220 KiB
36Accepted2/228ms1076 KiB
37Accepted2/228ms1156 KiB
38Accepted2/228ms980 KiB
39Accepted1/11ms316 KiB