Submission #375168

#	Time	Username	Problem	Language	Result	Execution time	Memory
375168		Kevin_Zhang_TW	Naan (JOI19_naan)	C++17	29 / 100	26 ms	4844 KiB

naan

#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;

int n, L;
ll v[MAX_N][MAX_N];

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 = inf * lft / dn, dn = inf;
		}
		return *this;
	}
};
frac operator * (frac a, frac b) { a.up *= b.up, a.dn *= b.dn, a.ref(); return a; }
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;
	a.chk();
	return a;
}

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 a.up * b.dn < b.up * a.dn; }
bool operator > (frac a, frac b) { return a.up * b.dn > b.up * a.dn; }
ostream& operator << (ostream& out, frac a) { return out << a.up << ' ' << 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];

	assert(max(n, L) < MAX_N);

	vector<int> od(n); iota(AI(od), 0);

	random_device rd;
	mt19937 gen(rd());

	int holan = 5000;
	do {
		// gen 
		bool ok = true;
		vector<frac> sep {{0, 1}};
		debug(AI(od));
		for (int x : od) {
			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];
				sep.pb(len + sep.back());
				continue;
			}
			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;
				break;
			}
			sep.pb( frac(atid) + need / v[x][atid] );
			DE(sep.back());
		}
		if (!ok) continue;
		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;
	} while (shuffle(AI(od), gen), holan--);

	cout << -1 << '\n';
}

Compilation message (stderr)

naan.cpp: In member function 'frac& frac::chk()':
naan.cpp:32:7: warning: unused variable 'num' [-Wunused-variable]
   32 |    ll num = up / dn, lft = up % dn;
      |       ^~~
naan.cpp: In function 'int32_t main()':
naan.cpp:15:20: warning: statement has no effect [-Wunused-value]
   15 | #define debug(...) 0
      |                    ^
naan.cpp:82:3: note: in expansion of macro 'debug'
   82 |   debug(AI(od));
      |   ^~~~~
naan.cpp:14:17: warning: statement has no effect [-Wunused-value]
   14 | #define DE(...) 0
      |                 ^
naan.cpp:91:4: note: in expansion of macro 'DE'
   91 |    DE(x, atid, lft, dn);
      |    ^~
naan.cpp:14:17: warning: statement has no effect [-Wunused-value]
   14 | #define DE(...) 0
      |                 ^
naan.cpp:111:4: note: in expansion of macro 'DE'
  111 |    DE(sep.back());
      |    ^~

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