227792026-01-15 18:27:04szjÓvodacpp17Wrong answer 33/5052ms4652 KiB
#include<bits/stdc++.h>
using namespace std;
struct kolyok
{
    int pulyanr, boges, valasz;
};
bool f(kolyok a, kolyok b)
{
    return a.boges > b.boges;
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    long long int n, m, i, siras=0, j=0;
    cin >> n >> m;
    int szerep[m], szerep2[m];
    vector<int> kiva(n, -1);
    stack<kolyok>st;
    kolyok t[n];
    for(i=0; i<m; i++)
    {
        cin >> szerep[i];
        szerep2[i]=szerep[i];
    }
    for(i=0; i<n; i++)
    {
        cin >> t[i].valasz;
        t[i].valasz--;
        t[i].pulyanr=i;
    }
    for(i=0; i<n; i++)cin >> t[i].boges;
    sort(t, t+n, f);
    for(i=0; i<n; i++)
    {
        if(szerep[t[i].valasz]>0)
        {
            kiva[t[i].pulyanr]=t[i].valasz;
            szerep[t[i].valasz]--;
            st.push(t[i]);
        }
    }
    queue<int>q;
    for(i=0; i<m; i++)if(szerep2[i]==szerep[i])q.push(i);
    for(i=0; i<n; i++)
    {
        for(j; j<m && szerep[j]==0; j++);
        if(kiva[t[i].pulyanr]==-1)
        {
            if(q.empty())
            {
                kiva[t[i].pulyanr]=j;
                szerep[j]--;
            }
            else
            {
                kiva[t[i].pulyanr]=q.front();
                szerep[q.front()]--;
                q.pop();
            }
            siras+=t[i].boges;
        }
    }
    while(!q.empty())
    {
        siras+=st.top().boges;
        kiva[st.top().pulyanr]=q.front();
        q.pop();
        st.pop();
    }
    cout << siras << endl;
    for(i=0; i<n; i++)cout << kiva[i]+1 << " ";
}
SubtaskSumTestVerdictTimeMemory
base33/50
1Accepted0/01ms500 KiB
2Accepted0/03ms564 KiB
3Accepted2/21ms316 KiB
4Accepted2/21ms500 KiB
5Accepted2/21ms316 KiB
6Accepted2/21ms316 KiB
7Wrong answer0/21ms316 KiB
8Accepted2/21ms316 KiB
9Accepted2/21ms316 KiB
10Accepted2/21ms316 KiB
11Accepted2/21ms316 KiB
12Accepted2/21ms316 KiB
13Wrong answer0/21ms316 KiB
14Accepted3/31ms316 KiB
15Wrong answer0/36ms668 KiB
16Wrong answer0/314ms1332 KiB
17Accepted3/320ms1756 KiB
18Wrong answer0/330ms2868 KiB
19Accepted3/335ms3048 KiB
20Accepted3/341ms3096 KiB
21Accepted3/346ms3640 KiB
22Wrong answer0/452ms4652 KiB