Submission #1132157

#	Time	Username	Problem	Language	Result	Execution time	Memory
1132157		StefanSebez	Carnival Tickets (IOI20_tickets)	C++20	14 / 100	664 ms	62708 KiB

tickets

#include <bits/stdc++.h>
#include "tickets.h"
using namespace std;
#define fi first
#define se second
#define pb push_back
#define ll long long
#define ld long double
const int N=1550;
int a[N][N];
vector<vector<int>>s;
int n,m;
ll izracunaj(vector<int>a){
    sort(a.begin(),a.end());
    //for(auto i:a) printf("%i ",i);printf("\n");
    ll sum1=0;for(auto i:a) sum1+=i;
    ll res=1e18,n=a.size();
    for(ll i=0,sum=0;i<n;i++){
        ll x=sum1-2*sum+a[i]*(2*i-n);
        //printf("%lld\n",x);
        res=min(res,x);
        sum+=a[i];
    }
    return res;
}
ll Izracunaj(){
    int maks=0;
    for(int i=0;i<n;i++) for(int j=0;j<m;j++) maks=max(maks,s[i][j]);
    vector<int>vrednosti[maks+1];
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            if(s[i][j]!=-1) vrednosti[s[i][j]].pb(a[i][j]);
        }
    }
    ll res=0;for(int i=0;i<=maks;i++) if(!vrednosti[i].empty()) res+=izracunaj(vrednosti[i]);
    return res;
}
long long find_maximum(int k, std::vector<std::vector<int>> a1){
    n=a1.size(),m=a1[0].size();
    for(int i=0;i<n;i++) for(int j=0;j<m;j++) a[i][j]=a1[i][j];
    for(int i=0;i<n;i++){
        vector<int>temp;
        for(int j=0;j<m;j++) temp.pb(-1);
        s.pb(temp);
    }
    /*vector<array<int,3>>nesto;
    for(int i=0;i<n;i++){
        int maks=0,ind=-1;
        for(int j=0;j<m;j++){
            if(a[i][j]>maks) maks=a[i][j],ind=j;
        }
        nesto.pb({maks,i,ind});
    }
    sort(nesto.begin(),nesto.end());
    for(int i=n-1;i>=n/2;i--) s[nesto[i][1]][nesto[i][2]]=0;
    for(int I=0;I<n/2;I++){
        int i=nesto[I][1];
        int mn=1e9,ind=-1;
        for(int j=0;j<m;j++){
            if(mn>a[i][j]) mn=a[i][j],ind=j;
        }
        s[i][ind]=0;
    }
    ll x=Izracunaj();vector<vector<int>>s1=s;
    for(int i=0;i<n;i++) for(int j=0;j<m;j++) s[i][j]=-1;
    nesto.clear();
    for(int i=0;i<n;i++){
        int mn=1e9,ind=-1;
        for(int j=0;j<m;j++){
            if(a[i][j]<mn) mn=a[i][j],ind=j;
        }
        nesto.pb({mn,i,ind});
    }
    sort(nesto.begin(),nesto.end());
    for(int i=0;i<n/2;i++) s[nesto[i][1]][nesto[i][2]]=0;
    for(int I=n/2;I<n;I++){
        int i=nesto[I][1];
        int maks=0,ind=-1;
        for(int j=0;j<m;j++){
            if(maks<a[i][j]) maks=a[i][j],ind=j;
        }
        s[i][ind]=0;
    }
    if(Izracunaj()<x) s=s1;*/
    /*bool izbrisan[n+10]={false};
    while(1){
        int d=0,maks=0,ind0=-1,mn=1e9,ind1=-1;
        pair<int,int>ubaci={-1,-1};
        for(int i=0;i<n;i++){
            if(izbrisan[i]) continue;
            if(a[i][m-1]-mn>d){
                d=a[i][m-1]-mn;
                ubaci={ind1,i};
            }
            if(maks-a[i][0]>d){
                d=maks-a[i][0];
                ubaci={i,ind0};
            }
            if(maks<a[i][m-1]) maks=a[i][m-1],ind0=i;
            if(mn>a[i][0]) mn=a[i][0],ind1=i;
        }
        if(ubaci.fi==-1) break;
        s[ubaci.fi][0]=0;
        s[ubaci.se][m-1]=0;
        izbrisan[ubaci.fi]=izbrisan[ubaci.se]=true;
    }*/
    int num[n+10][2],ind[n+10][2];
    memset(num,0,sizeof(num));
    memset(ind,0,sizeof(ind));
    for(int i=0;i<n;i++) ind[i][0]=0,ind[i][1]=m-1;
    for(int i=0;i<n;i++) for(int j=0;j<m;j++) num[i][a[i][j]]++;
    set<pair<int,int>>st[2];
    for(int i=0;i<n;i++) st[0].insert({num[i][0],i}),st[1].insert({num[i][1],i});
    for(int c=0;c<k;c++){
        int ct=n;bool was[n+10]={false};
        while(ct){
            if(st[0].size()&&st[1].size()){
                pair<int,int>x=*st[1].rbegin();st[1].erase(x);
                while(was[x.se]) x=*st[1].rbegin(),st[1].erase(x);
                num[x.se][1]--;
                s[x.se][ind[x.se][1]--]=c;
                was[x.se]=true;

                x=*st[0].rbegin();st[0].erase(x);
                while(was[x.se]) x=*st[0].rbegin(),st[0].erase(x);
                num[x.se][0]--;
                s[x.se][ind[x.se][0]++]=c;
                was[x.se]=true;
                ct-=2;
            }
            else if(st[0].empty()){
                pair<int,int>x=*st[1].rbegin();st[1].erase(x);
                while(was[x.se]) x=*st[1].rbegin(),st[1].erase(x);
                num[x.se][1]--;
                s[x.se][ind[x.se][1]--]=c;
                was[x.se]=true;
                ct--;
            }
            else{
                pair<int,int>x=*st[0].rbegin();st[0].erase(x);
                while(was[x.se]) x=*st[0].rbegin(),st[0].erase(x);
                num[x.se][0]--;
                s[x.se][ind[x.se][0]++]=c;
                was[x.se]=true;
                ct--;
            }
        }
        st[0].clear(),st[1].clear();
        for(int i=0;i<n;i++) st[0].insert({num[i][0],i}),st[1].insert({num[i][1],i});
    }
    allocate_tickets(s);
    return Izracunaj();
}

• Subtask 1: 0 / 11
• Subtask 2: 0 / 16
• Subtask 3: 14 / 14
• Subtask 4: 0 / 14
• Subtask 5: 0 / 12
• Subtask 6: 0 / 23
• Subtask 7: 0 / 10