Submission #1198311

#TimeUsernameProblemLanguageResultExecution timeMemory
1198311lrnnzCarnival Tickets (IOI20_tickets)C++17
27 / 100
291 ms51488 KiB
#include <bits/stdc++.h>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <iomanip>
#include "tickets.h"
using namespace std;
 
#define all(a) (a).begin(), (a).end()
#define sz(a) (int)(a).size()
#define ll long long
#define ld long double
#define ui uint64_t
#define cont(set, element) ((set).find(element) != (set).end())
 
/********* DEBUG *********/
 
template <typename T>
void outvec(const vector<T>& Z){
    for (const T& x : Z)
    cout << x << ' ';
    cout << "\n";
}
void printVariable(const any& var) {
    if (!var.has_value()) {
        cout << "null";
        return;
    }

    if (var.type() == typeid(int)) {
        cout << any_cast<int>(var);
    } else if (var.type() == typeid(double)) {
        cout << any_cast<double>(var);
    } else if (var.type() == typeid(float)) {
        cout << any_cast<float>(var);
    } else if (var.type() == typeid(char)) {
        cout << any_cast<char>(var);
    } else if (var.type() == typeid(bool)) {
        cout << (any_cast<bool>(var) ? "true" : "false");
    } else if (var.type() == typeid(string)) {
        cout << any_cast<string>(var);
    } else if (var.type() == typeid(const char*)) {
        cout << any_cast<const char*>(var);
    } else if (var.type() == typeid(long long)) {
        cout << any_cast<long long>(var);
    } else {
        cout << "[unknown type]";
    }
}

template<typename... Args>
void outval(Args... args) {
    vector<any> variables = {args...};
    
    for (size_t i = 0; i < variables.size(); ++i) {
        printVariable(variables[i]);
        if (i != variables.size() - 1) {
            cout << " ";
        }
    }
    cout << "\n";
}

#define nl "\n"
#define sp << " " <<

/********* DEBUG *********/

const ll MOD = 1000000007;
const ll inf = 1e18;
const ll mxN = 1000005;

ll find_maximum(int k, vector<vector<int>> x) {
	int n = x.size();
	int m = x[0].size();

    vector<ll> takeright(n);

    // [val = hi unused + hi used, idx]
    priority_queue<pair<ll,ll>> pq;
    for (int i = 0; i < n; i++){
        pq.push({x[i][k-1] + x[i][n-1] , i});
    }

    ll rs = 0;
    while (rs < n*k/2){
        auto [val, idx] = pq.top(); pq.pop();

        if (++takeright[idx] < k)
        pq.push({x[idx][k-1-takeright[idx]] + x[idx][n-1-takeright[idx]], idx});
        rs++;
    }

    vector<vector<int>> alloc(n, vector<int>(m, -1));
    //outvec(takeright);

    ll ans = 0, l = 0, r = 0, currl = 0, currr = 0;
    for (int i = 0; i < n; i++){
        ll cnt = 0;
        for (int j = 0; j < k-takeright[i]; j++){
            ans -= x[i][j];
            alloc[i][j] = l;

            if (++currl >= n/2)
                l++, currl = 0;
        }

        for (int j = m-1; j >= m-takeright[i]; j--){
            ans += x[i][j];
            alloc[i][j] = r;

            if (++currr >= n/2)
                r++, currr = 0;
        }
    }

    allocate_tickets(alloc);

    return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...