Submission #405918

#	Time	Username	Problem	Language	Result	Execution time	Memory
405918		ngrace	Carnival Tickets (IOI20_tickets)	C++14	11 / 100	2 ms	844 KiB

tickets

#include "tickets.h"
#include <bits/stdc++.h>
using namespace std;
 
#define p pair
#define pq priority_queue
#define v vector
#define card pair<int,pair<int,int>>

long long find_maximum(int k, v<v<int>> x) {
    long long out=0;
    int n=x.size();//num colours
    int m=x[0].size();//num of each colour
    
    pq<card> mi;
 
    v<v<card>> l(n);
    v<v<card>> s(n);
    
    v<v<int>> answer(n,v<int>(m,-1));
    
    for(int i=0;i<n;i++){//initilize long list
        pq<card> ma;
        for(int j=0;j<m;j++){
            ma.push({x[i][j],{i,j}});
            mi.push({-x[i][j],{i,j}});
        }
        for(int j=0;j<k;j++){
            card cur=ma.top();
            ma.pop();
            l[i].push_back(cur);
        }
    }
    int count=0;
    while(count<k*n/2){//transfer to short list
        card cur = mi.top();
        mi.pop();
        int colInd=cur.second.first;
        if(l[colInd].size()>0){
            l[colInd].pop_back();
            s[colInd].push_back({-cur.first,cur.second});
            count++;
        }
    }
    for(int i=0;i<k;i++){//allocate
        pq<pair<int,int>> numL;
        for(int j=0;j<n;j++){
            numL.push({l[j].size(),j});
        }
        int sm=0;
        int la=0;
        for(int j=0;j<n;j++){
            bool takeL=false;
            if(j<n/2){
                takeL=true;
            }

            int colInd=numL.top().second;
            numL.pop();
            if(takeL){
                card cur=l[colInd][l[colInd].size()-1];
                answer[colInd][cur.second.second]=i;
                out+=cur.first;
                l[colInd].pop_back();
                la++;
            }
            else{
                card cur=s[colInd][s[colInd].size()-1];
                answer[colInd][cur.second.second]=i;
                out-=cur.first;
                s[colInd].pop_back();
                sm++;
            }
        }
    }
	allocate_tickets(answer);
    return out;
}

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