//#undef KEV
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define pb emplace_back
#define AI(i) begin(i), end(i)
template<class T> bool chmin(T &a, T b) { return b < a && (a = b, true); }
template<class T> bool chmax(T &a, T b) { return a < b && (a = b, true); }
#ifdef KEV
#define DE(args...) kout("[ " + string(#args) + " ] = ", args)
void kout() { cerr << endl; }
template<class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ', kout(b...); }
template<class T> void debug(T l, T r) { while (l != r) cerr << *l << " \n"[next(l) == r], ++l; }
#else
#define DE(...) 0
#define debug(...) 0
#endif
const int MAX_N = 2005, inf = 1e9;
using opl = long long;
int n, L;
opl v[MAX_N][MAX_N];
using lll = __int128;
#define ll lll
opl gcd(lll a, lll b) { return gcd(opl(a), opl(b)); }
struct frac {
ll up, dn;
void ref() { ll g = gcd(up, dn); up/=g, dn/=g; }
frac(ll a, ll b) :
up(a), dn(b) {}
frac(ll v) :
up(v), dn(1) {}
void sca(ll v) { up *= v, dn *= v; }
frac& chk() {
if (dn > inf) {
ll num = up / dn, lft = up % dn;
up = (lll)inf * lft / dn + num * inf, dn = inf;
}
return *this;
}
};
frac operator * (frac a, frac b) { a.up *= b.up, a.dn *= b.dn, a.ref(); return a.chk(); }
frac operator * (frac a, ll v) { a.up *= v; return a.chk(); }
frac operator / (frac a, ll v) { a.dn *= v; return a.chk(); }
frac operator + (frac a, frac b) {
ll da = a.dn, db = b.dn, nd = da / gcd(da, db) * db;
a.sca(nd/da), b.sca(nd/db);
a.up += b.up;
return a.chk();
}
frac operator - (frac a, frac b) {
ll da = a.dn, db = b.dn, nd = da / gcd(da, db) * db;
a.sca(nd/da), b.sca(nd/db);
a.up -= b.up;
return a.chk();
}
frac& operator += (frac &a, frac b) { return a = a + b; }
frac& operator -= (frac &a, frac b) { return a = a - b; }
bool operator < (frac a, frac b) { return (lll)a.up * b.dn < (lll)b.up * a.dn; }
bool operator > (frac a, frac b) { return (lll)a.up * b.dn > (lll)b.up * a.dn; }
ostream& operator << (ostream& out, frac a) { return out << (opl)a.up << ' ' << (opl)a.dn; }
int32_t main() {
ios_base::sync_with_stdio(0), cin.tie(0);
cin >> n >> L;
for (int i = 0;i < n;++i)
for (int j = 0;j < L;++j)
cin >> v[i][j];
int holan = 5000;
random_device rd;
mt19937 gen(rd());
while (holan--) {
vector<int> res;
bool ok = true;
vector<frac> sep{{0, 1}};
auto geted = [&](int x) {
int sum = 0;
for (int j = 0;j < L;++j)
sum += v[x][j];
auto [up, dn] = sep.back();
int atid = up / dn, lft = dn - up % dn;
//DE(x, atid, lft, dn);
frac need{sum, n};
if (frac(lft, dn) * v[x][atid] > need) {
frac len = need / v[x][atid];
return len + sep.back();
}
need -= frac(lft, dn) * v[x][atid];
++atid;
while (atid < L) {
if (need < v[x][atid]) break;
need -= v[x][atid++];
}
if (atid == L && need > frac(0)) {
ok = false;
return frac(inf, 1);
}
frac ed = frac(atid) + need / v[x][atid];
return ed;
};
vector<int> od(n); iota(AI(od), 0);
shuffle(AI(od), gen);
debug(AI(od));
for (int pos = 0;pos < n;++pos) {
int id = -1;
frac ned(inf, 1);
for (int x : od) {
frac ed = geted(x);
if (chmin(ned, ed))
id = x;
}
if (id == -1) {
ok = false;
break;
}
res.pb(id);
sep.pb(ned);
DE(id, ned);
debug(AI(od));
od.erase(find(AI(od), id));
debug(AI(od));
}
if (!ok) continue;
for (int i = 1;i < n;++i)
cout << sep[i] << '\n';
for (int i = 0;i < n;++i)
cout << res[i]+1 << " \n"[i+1==n];
return 0;
}
//cout << -1 << '\n';
}
Compilation message
naan.cpp: In function 'int32_t main()':
naan.cpp:16:20: warning: statement has no effect [-Wunused-value]
16 | #define debug(...) 0
| ^
naan.cpp:123:3: note: in expansion of macro 'debug'
123 | debug(AI(od));
| ^~~~~
naan.cpp:15:17: warning: statement has no effect [-Wunused-value]
15 | #define DE(...) 0
| ^
naan.cpp:142:4: note: in expansion of macro 'DE'
142 | DE(id, ned);
| ^~
naan.cpp:16:20: warning: statement has no effect [-Wunused-value]
16 | #define debug(...) 0
| ^
naan.cpp:144:4: note: in expansion of macro 'debug'
144 | debug(AI(od));
| ^~~~~
naan.cpp:16:20: warning: statement has no effect [-Wunused-value]
16 | #define debug(...) 0
| ^
naan.cpp:146:4: note: in expansion of macro 'debug'
146 | debug(AI(od));
| ^~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
364 KB |
Output is correct |
| 2 |
Correct |
1 ms |
364 KB |
Output is correct |
| 3 |
Correct |
1 ms |
364 KB |
Output is correct |
| 4 |
Correct |
1 ms |
364 KB |
Output is correct |
| 5 |
Correct |
1 ms |
364 KB |
Output is correct |
| 6 |
Correct |
1 ms |
364 KB |
Output is correct |
| 7 |
Correct |
1 ms |
364 KB |
Output is correct |
| 8 |
Correct |
1 ms |
364 KB |
Output is correct |
| 9 |
Correct |
1 ms |
364 KB |
Output is correct |
| 10 |
Correct |
1 ms |
364 KB |
Output is correct |
| 11 |
Incorrect |
593 ms |
364 KB |
Unexpected end of file - int64 expected |
| 12 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
364 KB |
Output is correct |
| 2 |
Correct |
1 ms |
364 KB |
Output is correct |
| 3 |
Correct |
1 ms |
384 KB |
Output is correct |
| 4 |
Correct |
1 ms |
492 KB |
Output is correct |
| 5 |
Correct |
1 ms |
364 KB |
Output is correct |
| 6 |
Correct |
1 ms |
364 KB |
Output is correct |
| 7 |
Correct |
1 ms |
364 KB |
Output is correct |
| 8 |
Correct |
1 ms |
364 KB |
Output is correct |
| 9 |
Correct |
1 ms |
492 KB |
Output is correct |
| 10 |
Correct |
2 ms |
492 KB |
Output is correct |
| 11 |
Correct |
1 ms |
364 KB |
Output is correct |
| 12 |
Correct |
1 ms |
364 KB |
Output is correct |
| 13 |
Correct |
1 ms |
364 KB |
Output is correct |
| 14 |
Correct |
2 ms |
492 KB |
Output is correct |
| 15 |
Correct |
1 ms |
364 KB |
Output is correct |
| 16 |
Correct |
2 ms |
492 KB |
Output is correct |
| 17 |
Correct |
2 ms |
492 KB |
Output is correct |
| 18 |
Incorrect |
1586 ms |
492 KB |
Unexpected end of file - int64 expected |
| 19 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
364 KB |
Output is correct |
| 2 |
Correct |
1 ms |
364 KB |
Output is correct |
| 3 |
Correct |
1 ms |
364 KB |
Output is correct |
| 4 |
Correct |
1 ms |
364 KB |
Output is correct |
| 5 |
Correct |
1 ms |
364 KB |
Output is correct |
| 6 |
Correct |
1 ms |
364 KB |
Output is correct |
| 7 |
Correct |
1 ms |
364 KB |
Output is correct |
| 8 |
Correct |
1 ms |
364 KB |
Output is correct |
| 9 |
Correct |
1 ms |
364 KB |
Output is correct |
| 10 |
Correct |
1 ms |
364 KB |
Output is correct |
| 11 |
Incorrect |
593 ms |
364 KB |
Unexpected end of file - int64 expected |
| 12 |
Halted |
0 ms |
0 KB |
- |
