Submission #421700

#	Time	Username	Problem	Language	Result	Execution time	Memory
421700		idk321	Carnival Tickets (IOI20_tickets)	C++17	11 / 100	2 ms	588 KiB

tickets

#include "tickets.h"
#include <vector>

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const ll INF = 1000000000000000000LL;

long long find_maximum(int k, std::vector<std::vector<int>> x) {
	int n = x.size();
	int m = x[0].size();
	vector<vector<int>> answer;
    answer.resize(n, vector<int>(m, -1));
    if (m == 1)
    {

        for (int i = 0; i < n; i++) answer[i][0] = 0;
        allocate_tickets(answer);
        vector<int> values(n);
        for (int i = 0; i < n; i++)
        {
            values[i] = x[i][0];
        }
        sort(values.begin(), values.end());

        ll res = 0;
        int mid = (0 + n - 1) / 2;
        for (int i = 0; i < n; i++)
        {
            res += abs(values[mid] - values[i]);
        }

        return res;
    }

    vector<vector<array<int, 2>>> valAndInd(n, vector<array<int, 2>>(m));
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j  < m; j++)
        {
            valAndInd[i][j] = {x[i][j], j};
        }
    }
    for (int i = 0; i < n; i++) sort(valAndInd[i].begin(), valAndInd[i].end());


    int big = 0;
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++) big = max(big, x[i][j]);
    }

    if (big <= 1)
    {
        int turn = 0;
        int res = k;
        for (int j = 0; j < m; j++)
        {
            int sum = 0;
            for (int i = 0; i < n; i++)
            {
                sum += valAndInd[i][j][0];
            }
            if (sum == 0)
            {
                for (int i = 0; i < n; i++)
                {
                    answer[i][valAndInd[i][j][1]] = turn;
                }
                turn++;
                res--;
            }
            if (sum == n)
            {
                for (int i = 0; i < n; i++)
                {
                    answer[i][valAndInd[i][j][1]] = turn;
                }
                turn++;
                res--;
            }
            if (turn == k) break;
        }

        int cur = 0;

        vector<vector<int>> emptyIn(n);
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < m; j++)
            {
                if (answer[i][j] == -1) emptyIn[i].push_back(j);
            }
        }

        for (; turn < k; turn++)
        {
            for (int i = 0; i < n; i++)
            {
                answer[i][emptyIn[i].back()] = turn;
                emptyIn[i].pop_back();
            }
        }


        allocate_tickets(answer);
        return res;
    }

    /*
    if (k == 1)
    {
        array<int, 3> best = {-1, -1, -1};

        vector<array<int, 2>> byVal;
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < m; j++)
            {
                byVal.push_back({x[i][j], i});
            }
        }
        sort(byVal.begin(), byVal.end());
        vector<array<int, 2>> order;
        for (int i = 0; i < n; i++)
        {
            order.push_back({x[i].front(), i});
        }
        ll cres = 0;
        for (int i = 0; i < n; i++)
        {
            cres += x[i].back() - byVal[0][0];
        }
        ll res = INF;
        int take = 0;
        for (auto val : byVal)
        {
            int num = val[0];
            while (take < n && order[take].front() <= num)
            {
                cres -=
                take++;
            }
        }
    } */



	return 1;
}

/*
3 4 3
1 0 1 1
0 1 0 0
0 0 1 1
*/

Compilation message (stderr)

tickets.cpp: In function 'long long int find_maximum(int, std::vector<std::vector<int> >)':
tickets.cpp:86:13: warning: unused variable 'cur' [-Wunused-variable]
   86 |         int cur = 0;
      |             ^~~

• 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