Submission #375253

#	Time	Username	Problem	Language	Result	Execution time	Memory
375253		Kevin_Zhang_TW	Naan (JOI19_naan)	C++17	0 / 100	3 ms	512 KiB

naan

//#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
 
lll gcd(lll a, lll b) { 
	return b == 0 ? a : gcd(b, a % b);
}
 
struct frac {
	ll up, dn;
	void ref() { ll g = gcd(up, dn); if (g == 0) return; 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() {
		ref();
		if (dn > inf) {
			ll num = up / dn, lft = up - num * 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; }
 
ll sumon[MAX_N];

int mxpos[MAX_N], mxs[MAX_N];
 
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], sumon[i] += v[i][j];

	for (int i = 0;i < n;++i) {
		for (int j = 0;j < L;++j)
			if (chmax(mxs[i], (int)v[i][j]))
				mxpos[i] = j;
	}
//	assert(n == L);
//
//	for (int i = 0;i < n;++i)
//		DE(i, mxpos[i]);
//
//	DE((opl)sumon[19], (int)mxs[19], (opl)sumon[19] / n);

	vector<int> vis(n);

	bool ok = true;
	vector<frac> sep {{0, 1}};
	vector<int> od;

	for (int i = 0;i < n;++i) {
		frac nl = frac(inf, 1);

		int nid = -1;

		for (int x = 0;x < n;++x) if (!vis[x]) {
			ll sum = sumon[x];
 
			auto [up, dn] = sep.back();
 
			ll atid = up / dn, lft = dn - (up % dn);
 
			frac need{sum, n};
 
			if (frac(lft, dn) * v[x][atid] > need) {
				frac len = need / v[x][atid];
				if (chmin(nl, len + sep.back()))
					nid = x;

				continue;
			}
			need -= frac(lft, dn) * v[x][atid++];
			while (atid < L) {
				if (need < v[x][atid]) break;
				need -= v[x][atid++];
			}
			if (atid == L && need > frac(0)) {
				continue;
			}
			if (chmin(nl, frac(atid) + need / v[x][atid]))
				nid = x;
		}
		if (nid == -1) {
			break;
		}
		od.pb(nid);
		sep.pb(nl);
		DE(i, nid, nl, (opl)(nl.up/nl.dn));
		vis[nid] = true;
	}

	DE(od.size());
	if (od.size() != n) return 0;

	for (int i = 1;i < n;++i)
		cout << sep[i] << '\n';
	for (int i = 0;i < n;++i)
		cout << od[i]+1 << " \n"[i+1==n];
	return 0;
}

Compilation message (stderr)

naan.cpp: In function 'int32_t main()':
naan.cpp:15:17: warning: statement has no effect [-Wunused-value]
   15 | #define DE(...) 0
      |                 ^
naan.cpp:140:3: note: in expansion of macro 'DE'
  140 |   DE(i, nid, nl, (opl)(nl.up/nl.dn));
      |   ^~
naan.cpp:15:17: warning: statement has no effect [-Wunused-value]
   15 | #define DE(...) 0
      |                 ^
naan.cpp:144:2: note: in expansion of macro 'DE'
  144 |  DE(od.size());
      |  ^~
naan.cpp:145:16: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  145 |  if (od.size() != n) return 0;
      |      ~~~~~~~~~~^~~~
naan.cpp:99:7: warning: unused variable 'ok' [-Wunused-variable]
   99 |  bool ok = true;
      |       ^~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 24
• Subtask 3: 0 / 71