Submission #333804

#	Time	Username	Problem	Language	Result	Execution time	Memory
333804		LucaDantas	Hiring (IOI09_hiring)	C++17	50 / 100	458 ms	28524 KiB

hiring

#include<bits/stdc++.h>
using namespace std;

using ll = long long;
using pii = pair<int,int>;

#define ff first
#define ss second

constexpr int maxn = 5e5+10, logn = 20;

struct Fraction
{
	int p, q, id;
	Fraction() {}
	Fraction(int a, int b, int c) : p(a), q(b), id(c) {}
	bool operator<(Fraction b) { return 1ll*p*b.q < 1ll*q*b.p; }
	ll operator*(ll a) {return (1ll * a * q) / p;}
} val[maxn];

struct BIT
{
	ll bit[maxn];
	void upd(int x, int v) {
		for(; x < maxn; x += x&-x)
			bit[x] += v;
	}
	ll query(int x) {
		ll ans = 0;
		for(; x > 0; x -= x&-x)
			ans += bit[x];
		return ans;
	}
	int bs(ll& val) {
		int pos = 0;
		for(int l = logn; l >= 0; l--) {
			if(pos + (1 << l) >= maxn) continue;
			if(bit[pos + (1 << l)] <= val)
				pos += (1 << l), val -= bit[pos];
		}
		return pos;
	}
	void clear() {
		for(int i = 0; i < maxn; i++)
			bit[i] = 0;
	}
} bit, ligados;

int pos[maxn], state[maxn];
pii opa[maxn];

int main() {
	int n; ll w; scanf("%d %lld", &n, &w);
	for(int i = 0, a, b; i < n; i++)
		scanf("%d %d", &a, &b), val[i] = Fraction(a, b, i), opa[i] = {b, i};
	sort(opa, opa+n);
	for(int i = 0; i < n; i++)
		pos[opa[i].ss] = i+1;
	sort(val, val+n);

	ll ans = 0, sobra = 0;
	for(int i = 0; i < n; i++) {
		bit.upd(pos[val[i].id], val[i].q);
		ligados.upd(pos[val[i].id], 1);

		ll pay = val[i] * w;
		
		int p = bit.bs(pay);
		int here = ligados.query(p);
		
		if(here == ans)
			sobra = max(sobra, (pay*val[i].p)/val[i].q);
		else if(here > ans) ans = here, sobra = (pay*val[i].p)/val[i].q;
	}
	bit.clear();
	ligados.clear();

	for(int i = 0; i < n; i++) {
		bit.upd(pos[val[i].id], val[i].q);
		ligados.upd(pos[val[i].id], 1);
		state[pos[val[i].id]] = 1;

		ll pay = val[i] * w;
		
		int p = bit.bs(pay);
		int here = ligados.query(p);
		
		if(here == ans && (pay*val[i].p)/val[i].q == sobra) {
			printf("%lld\n", ans);
			for(int j = 1; j <= p; j++)
				if(state[j])
					printf("%d\n", j);
			return 0;
		}
	}

	assert(0);

	// To achando q vai dar errado por causa de precisão mas vamo testar

	// printf("%lld\n", ans);
	// puts("TESTANDO SE A RESPOSTA TÁ CERTA DPS RECUPERO QUAL FOI");
}

Compilation message (stderr)

hiring.cpp: In function 'int main()':
hiring.cpp:53:20: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   53 |  int n; ll w; scanf("%d %lld", &n, &w);
      |               ~~~~~^~~~~~~~~~~~~~~~~~~
hiring.cpp:55:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   55 |   scanf("%d %d", &a, &b), val[i] = Fraction(a, b, i), opa[i] = {b, i};
      |   ~~~~~^~~~~~~~~~~~~~~~~

• Subtask 1: 15 / 30
• Subtask 2: 14 / 28
• Subtask 3: 3 / 6
• Subtask 4: 3 / 6
• Subtask 5: 3 / 6
• Subtask 6: 3 / 6
• Subtask 7: 3 / 6
• Subtask 8: 3 / 6
• Subtask 9: 3 / 6