Submission #303465

#	Time	Username	Problem	Language	Result	Execution time	Memory
303465		Vimmer	Carnival Tickets (IOI20_tickets)	C++14	11 / 100	2 ms	640 KiB

tickets

#include <bits/stdc++.h>
#include "tickets.h"
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>

#define N 1005
#define PB push_back
#define sz(x) int(x.size())
#define P 31
#define F first
#define M ll(1e9 + 7)
#define S second
#define all(x) x.begin(), x.end()
#define endl '\n'

//#pragma GCC optimize("unroll-loops")
//#pragma GCC optimize("-O3")
//#pragma GCC optimize("Ofast")
//#pragma GCC optimize("fast-math")
//#pragma GCC optimize("no-stack-protector")

using namespace std;
//using namespace __gnu_pbds;

typedef long long ll;

//typedef tree<int, null_type, less_equal <int>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;

ll mlt(ll a, ll b) {return (a * b) % M;}
ll sm(ll a, ll b) {return (a + b) % M;}

ll calc(vector <vector <int> > &x, ll md)
{
    ll sum = 0;

    for (int i = 0; i < sz(x); i++) sum += abs(md - x[i][0]);

    return sum;
}

ll clc(vector <int> &x, ll md)
{
    ll sum = 0;

    for (int i = 0; i < sz(x); i++) sum += abs(md - x[i]);

    return sum;
}

ll calcer(vector <vector <int> > &x, ll md)
{
    ll ans = 0;

    vector <int> xt; xt.clear();

    for (int i = 0; i < sz(x); i++)
        if (abs(md - x[i][0]) > abs(md - x[i].back())) xt.PB(x[i][0]);
         else xt.PB(x[i].back());

    sort(xt.begin(), xt.end());

    ll l = 0, r = 1e9;

    while (l < r)
    {
        ll mdl = (l + r) / 2;

        ll mdr = mdl + 1;

        if (clc(xt, mdl) > clc(xt, mdr)) l = mdr;
          else r = mdl;
    }

    return clc(xt, l);
}

ll find_maximum(int k, vector<vector<int> > x)
{
	int n = sz(x);

	int m = sz(x[0]);

    if (m == 1)
    {
        ll l = 0, r = 1e9;

        while (l < r)
        {
            ll mdl = (l + r) / 2;

            ll mdr = mdl + 1;

            if (calc(x, mdl) < calc(x, mdr)) r = mdl;
              else l = mdr;
        }

        ll ans = calc(x, l);

        for (int i = 0; i < n; i++)
            x[i][0] = 0;

        allocate_tickets(x);

        return ans;
    }
    else
    {
        ll ans = 0;

        vector <vector <int> > r;

        r.resize(n);

        int mn = 1e9, mx = 0;

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

        for (int i = 0; i < n; i++)
        {
            r[i].resize(m);

            for (int j = 0; j < m; j++) r[i][j] = -1;
        }

        ll l = mn, rt = mx;

        while (l < rt)
        {
            ll mdl = (l + rt) / 2;

            ll mdr = mdl + 1;

            if (calcer(x, mdl) > calcer(x, mdr)) rt = mdl;
              else l = mdr;
        }

        ans = calcer(x, l);

        vector <int> xt; xt.clear();

        for (int i = 0; i < sz(x); i++)
            if (abs(l - x[i][0]) > abs(l - x[i].back())) r[i][0] = 0;
             else r[i][sz(r[i]) - 1] = 0;

        allocate_tickets(r);

        return ans;
    }
}


//int main()
//{
////    freopen("help.in", "r", stdin); freopen("help.out", "w", stdout);
//
//    ios_base::sync_with_stdio(0); istream::sync_with_stdio(0); cin.tie(0); cout.tie(0);
//
//    cout << find_maximum(1, {{0}, {10}, {3}, {5}, {8}, {1}}) << endl;
//}

Compilation message (stderr)

tickets.cpp: In function 'll calcer(std::vector<std::vector<int> >&, ll)':
tickets.cpp:52:8: warning: unused variable 'ans' [-Wunused-variable]
   52 |     ll ans = 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