제출 #793040

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
793040		Khizri	카니발 티켓 (IOI20_tickets)	C++17	41 / 100	3057 ms	71384 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 a,int b){
    ll x=0,y=0;
    for(int i=0;i<=mid[a];i++){
        x+=vt[a][i];
    }
    for(int i=0;i<=mid[b];i++){
        y+=vt[b][i];
    }
    return vt[a][r[a]]+x>vt[b][r[b]]+y;
}
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()/2;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[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]]++;
        }
        sum+=b-a;
	}
	allocate_tickets(ans);
	return sum;
}
/*
4 4 3
0 0 1 1
0 1 1 1
0 0 1 1
0 1 1 1
*/

컴파일 시 표준 에러 (stderr) 메시지

tickets.cpp: In function 'long long int find_maximum(int, std::vector<std::vector<int> >)':
tickets.cpp:52: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]
   52 |         for(int i=0;i<v.size()/2;i++){
      |                     ~^~~~~~~~~~~

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