#include "tickets.h"
#include <vector>
#include <bits/stdc++.h>
#define ll long long
#define ull unsigned ll
#define f first
#define s second
#define pii pair<int,int>
#define pll pair<ll,ll>
#define pb push_back
#define epb emplace_back
using namespace std;
long long find_maximum(int k, std::vector<std::vector<int>> x) {
int n = x.size();
int m = x[0].size();
ll cursum = 0;
set <pll> st;
ll curl[n], curr[n];
vector <vector<int> > al;
al.resize(n);
for(int i = 0; i < n; i++){
al[i].resize(m);
for(int j = 0; j < m; j++) al[i][j] = -1;
for(int j = m - k; j < m; j++)al[i][j] = j - m + k;
}
for(int i = 0; i < n; i++){
curr[i] = m - k;
curl[i] = 0;
st.insert({(ll)x[i][m - k] + (ll)x[i][0], i});
}
// ll cursum = 0;
for(int i = 0; i < n; i++){
for(int j = m - k; j < m; j++)cursum += (ll)x[i][j];
}
int raod = n * k / 2;
while(raod--){
pll now = *st.begin();
st.erase(st.begin());
cursum -= now.f;
int ind = now.s;
al[ind][curl[ind]] = al[ind][curr[ind]];
if(curr[ind] != curl[ind])al[ind][curr[ind]] = -1;
curl[ind]++;
curr[ind]++;
//cout << ind << "\n";
int ind1 = curl[ind], ind2 = curr[ind];
if(ind2 < m)
st.insert({x[ind][ind1] + x[ind][ind2], ind});
}
//cout << 1;
allocate_tickets(al);
return cursum;
}
