155382025-02-20 11:36:09MatematikaMaximális szorzat (50 pont)cpp17Accepted 50/5082ms2348 KiB
#include <bits/stdc++.h>

using namespace std;
#define int long long
const int MOD=1000000007;

signed main()
{
    int n,k,m,kezdes=-1,negativ=0,ossz=1; cin >> n >> k >> m;
    priority_queue<int, vector<int>, greater<int>> pq;
    vector<int> vec(n);
    for(auto& z:vec)
    {
        cin >> z;
        if(z<0)
            negativ++;
    }
    //cerr << negativ << "\n";
    if(m>negativ)
    {
        cout << "-1";
        return 0;
    }
    sort(vec.begin(),vec.end());
    for(int i=0; i<n && vec[i]<0; i++)
        kezdes=i;
    //cerr << kezdes << "\n";
    if(negativ==m && negativ==n)
    {
        for(auto& z:vec)
            if(z<-1)
                pq.push(z);
        while(k>0)
        {
            if(pq.empty())
            {
                cout << "-1";
                return 0;
            }
            int x=pq.top();
            //cerr << x << "\n";
            pq.pop();
            x++;
            if(x!=-1)
                pq.push(x);
            k--;
        }
        while(!pq.empty())
        {
            ossz*=pq.top();
            ossz%=MOD;
            pq.pop();
        }
        for(auto& z:vec)
            if(z>=0)
            {
                ossz*=z;
                ossz%=MOD;
            }
        cout << ossz;
        return 0;
    }
    else
    {
        for(int i=0; i<negativ-m; i++)
        {
            k+=vec[kezdes-i];
            vec[kezdes-i]=0;
        }
        /*for(auto z:vec)
            cerr << z << " ";*/
        if(k<0)
        {
            cout << "-1";
            return 0;
        }
        for(auto& z:vec)
            if(z>=0)
                pq.push(z);
        while(k>0 && !pq.empty())
        {
            int x=pq.top();
            //cerr << x << "\n";
            pq.pop();
            x++;
            pq.push(x);
            k--;
        }
        while(!pq.empty())
        {
            ossz*=pq.top();
            ossz%=MOD;
            pq.pop();
        }
        for(auto& z:vec)
            if(z<0)
            {
                ossz*=z;
                ossz%=MOD;
            }
        cout << ossz;
        return 0;
    }
}
/*4 2 4
-3 -1 -1 -1
*/
SubtaskSumTestVerdictTimeMemory
base50/50
1Accepted0/01ms316 KiB
2Accepted0/01ms316 KiB
3Accepted0/01ms316 KiB
4Accepted0/01ms316 KiB
5Accepted0/04ms564 KiB
6Accepted2/21ms316 KiB
7Accepted2/21ms316 KiB
8Accepted2/21ms316 KiB
9Accepted2/21ms316 KiB
10Accepted2/27ms568 KiB
11Accepted2/271ms2344 KiB
12Accepted1/178ms2340 KiB
13Accepted1/11ms316 KiB
14Accepted1/17ms584 KiB
15Accepted1/119ms564 KiB
16Accepted1/141ms1064 KiB
17Accepted1/120ms580 KiB
18Accepted1/112ms1444 KiB
19Accepted1/171ms2220 KiB
20Accepted1/157ms2348 KiB
21Accepted1/182ms2340 KiB
22Accepted1/126ms1836 KiB
23Accepted1/175ms1712 KiB
24Accepted1/178ms2340 KiB
25Accepted2/21ms316 KiB
26Accepted2/24ms316 KiB
27Accepted2/246ms1200 KiB
28Accepted1/146ms1436 KiB
29Accepted2/217ms800 KiB
30Accepted1/159ms1076 KiB
31Accepted1/141ms1140 KiB
32Accepted2/22ms316 KiB
33Accepted2/259ms1076 KiB
34Accepted1/161ms1196 KiB
35Accepted2/263ms1076 KiB
36Accepted2/261ms1076 KiB
37Accepted2/259ms1196 KiB
38Accepted2/259ms1076 KiB
39Accepted1/12ms316 KiB