Submission #793005

#	Time	Username	Problem	Language	Result	Execution time	Memory
793005		Khizri	카니발 티켓 (IOI20_tickets)	C++17	0 / 100	1 ms	300 KiB

tickets

#include "tickets.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define F first
#define S second
#define INF 1e18
#define all(v) (v).begin(),(v).end()
#define rall(v) (v).rbegin(),(v).rend()
#define pii pair<int,int>
#define pll pair<ll,ll>
#define OK cout<<"Ok"<<endl;
#define MOD (ll)(1e9+7)
const int mxn=1500+5;
int n,m,l[mxn],mid[mxn],r[mxn],sz[mxn];
vector<vector<int>>vt;
bool cmp(int i,int j){
    return vt[i][r[i]]+vt[i][mid[i]]>vt[j][r[j]]+vt[j][mid[j]];
}
bool cmp2(int i,int j){
    return sz[i]>sz[j];
}
long long find_maximum(int k, vector<vector<int>> x) {
    vt=x;
    n = x.size();
	m = x[0].size();
	vector<std::vector<int>> ans(n,vector<int>(m,-1));
	ll sum=0;
	vector<int>idx;
	for(int i=0;i<n;i++){
        l[i]=0;
        r[i]=m-1;
        mid[i]=k-1;
        idx.pb(i);
	}
	if(k==m){
        vector<pii>v;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                v.pb({vt[i][j],i});
            }
        }
        sort(all(v));
        for(int i=0;i<v.size();i++){
            sz[v[i].S]++;
        }
        for(int id=1;id<=k;id++){
            sort(all(idx),cmp2);
            ll a=0,b=0;
            for(int i=0;i<n/2;i++){
                a+=vt[idx[i]][l[idx[i]]];
                ans[idx[i]][l[idx[i]]]=id-1;
                sz[l[idx[i]]]--;
                l[idx[i]]++;
            }
            for(int i=n/2;i<n;i++){
                b+=vt[idx[i]][r[idx[i]]];
                ans[idx[i]][r[idx[i]]]=id-1;
                r[idx[i]]--;
            }
            sum+=b-a;
        }
        allocate_tickets(ans);
        return sum;
	}
	for(int id=1;id<=k;id++){
        sort(all(idx),cmp);
        ll a=0,b=0;
        for(int i=0;i<n/2;i++){
            b+=vt[idx[i]][r[idx[i]]];
            ans[idx[i]][r[idx[i]]]=id-1;
            r[idx[i]]--;
            mid[idx[i]]--;
        }
        for(int i=n/2;i<n;i++){
            a+=vt[idx[i]][l[idx[i]]];
            ans[idx[i]][l[idx[i]]]=id-1;
            l[idx[i]]++;
        }
        cout<<b<<' '<<a<<endl;
        sum+=b-a;
	}
	allocate_tickets(ans);
	return sum;
}
/*
4 4 4
0 0 0 1
0 0 1 1
0 0 0 1
0 1 1 1
*/

Compilation message (stderr)

tickets.cpp: In function 'long long int find_maximum(int, std::vector<std::vector<int> >)':
tickets.cpp:45:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   45 |         for(int i=0;i<v.size();i++){
      |                     ~^~~~~~~~~

• Subtask 1: 0 / 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