제출 #370979

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
370979		shivensinha4	로봇 (IOI13_robots)	C++17	28 / 100	1130 ms	26092 KiB

robots

#include "bits/stdc++.h"
#include"robots.h"
using namespace std;
#define for_(i, s, e) for (int i = s; i < (int) e; i++)
#define for__(i, s, e) for (ll i = s; i < e; i++)
typedef long long ll;
typedef vector<int> vi;
typedef array<int, 2> ii;
//#define endl '\n'

// Ask for sum 1 -> n for full (one based indexing)
class BIT {
	private: vector<ll> tree; int n = 1e6;
	int LSOne(int x) {
		return x&(-x);
	}

	public:
		void build(int x) {
			n = x;
			tree.resize(n+1);
		}
		ll sum(int a) {
			ll sum = 0;
			for (; a > 0; a -= LSOne(a)) sum += tree[a];
			return sum;
		}
		ll sum(int a, int b) {
			return sum(b) - (a == 1 ? 0 : sum(a-1));
		}
		void update(int p, ll v) {
			for (; p < n+1; p += LSOne(p)) tree[p] += v;
		}
};

int a, b, t;
vi wlim, slim;
vector<ii> toys;
BIT tree;

bool check(ll box) {
	vi left, done;
	
	for (int i = t-1; i >= 0; i--) {
//		cout << i << endl;
		int pt = lower_bound(slim.begin(), slim.end(), toys[i][1]) - slim.begin();
		if (pt < b and tree.sum(pt+1, b) < box * (b-pt)) {
			tree.update(pt+1, 1);
			done.push_back(pt);
		} else left.push_back(i);
//		cout << "---" << endl;
	}
	
	for (auto &i: done) tree.update(i+1, -1);
	done.clear();
	
	bool valid = true;
	
	for (auto i: left) {
		int pt = lower_bound(wlim.begin(), wlim.end(), toys[i][0]) - wlim.begin();
		if (pt < a and tree.sum(pt+1, a) < box * (a-pt)) {
			tree.update(pt+1, 1);
			done.push_back(pt);
		} else {
			valid = false;
			break;
		}
	}
	
	for (auto &i: done) tree.update(i+1, -1);
	
	return valid;
}


int putaway(int A, int B, int T, int X[], int Y[], int W[], int S[]) {
    a = A, b = B, t = T;
	wlim.reserve(A); slim.reserve(B); toys.reserve(T);
    for_(i, 0, a) {
		wlim.push_back(X[i]);
		wlim[i] -= 1;
    }
	for_(i, 0, b) {
		slim.push_back(Y[i]);
		slim[i] -= 1;
    }
	for_(i, 0, t) toys.push_back({W[i], S[i]});
	sort(wlim.begin(), wlim.end());
	sort(slim.begin(), slim.end());
	sort(toys.begin(), toys.end());
	
	for_(i, 0, t) if ((a == 0 or toys[i][0] > wlim.back()) and (b == 0 or toys[i][1] > slim.back())) return -1;
	
	tree.build(max(a, b));
	int l = 0, r = T+1, ans = r;
	while (l < r) {
		int mid = (l+r)/2;
		bool fin = check(mid);
//		cout << "checking " << mid << " " << fin << endl;
		if (fin) {
			ans = r = mid;
		} else l = mid+1;
	}
	
//	cout << "fin: " << ans << endl;
	return ans;
}

• Subtask 1: 14 / 14
• Subtask 2: 14 / 14
• Subtask 3: 0 / 25
• Subtask 4: 0 / 37
• Subtask 5: 0 / 10