#include "tickets.h"
#include <bits/stdc++.h>
#define pb push_back
#define all(x) begin(x),end(x)
#define SZ(x) (int)(x).size()
#define mem(x) memset(&x,y,sizeof(x))
#define ll long long
using namespace std;
ll find_maximum(int k, vector<vector<int>> x) {
int n=SZ(x),m=SZ(x[0]);
vector<vector<int>> answer(n,vector<int>(m,-1));
vector<array<int,3>> vec;
ll s=0;
for(int i=0; i<n; i++) {
for(int j=0; j<k; j++) {
s+=x[i][j+m-k];
vec.pb({-x[i][j]-x[i][j+m-k],-j,i});
}
}
sort(all(vec));reverse(all(vec));
set<array<int,2>> rows;
for(int i=0; i<k*n/2; i++) {
s+=vec[i][0];
rows.insert({-vec[i][1],vec[i][2]});
}
for(int i=k-1; i>=0; i--) {
assert(SZ(rows)>=n/2);
auto it=prev(end(rows));
vector<array<int,2>> todel;
for(int j=0; j<n/2; j++) {
int x=(*it)[1],y=(*it)[0];
answer[x][y]=0;
todel.pb(*it);
it--;
}
for(auto &j:todel)
rows.erase(j);
}
for(int i=0; i<n; i++) {
int cnt=0;
for(int j=0; j<m; j++) {
if(answer[i][j]!=-1) {
answer[i][j]+=cnt;
cnt++;
}
}
for(int j=m-1; j>m-1-(k-cnt); j--) {
answer[i][j]=k-1+j-(m-1);
}
}
allocate_tickets(answer);
return s;
}
