Submission #1067526

#	Time	Username	Problem	Language	Result	Execution time	Memory
1067526		Ignut	커다란 상품 (IOI17_prize)	C++17	20 / 100	73 ms	19548 KiB

prize

// Ignut

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

vector<int> ask(int i);

int Q = 0;

struct SegmentTree {
	vector<int> t;
	void Init(int n) {
		t.assign(4 * n, 0);
	}
	void Add(int v, int l, int r, int pos) {
		t[v] ++;
		if (l == r) return;
		int mid = l + (r - l) / 2;
		if (pos <= mid)
			Add(v * 2, l, mid, pos);
		else
			Add(v * 2 + 1, mid + 1, r, pos);
	}
	int Sum(int v, int l, int r, int ql, int qr) {
		if (l > qr || ql > r)
			return 0;
		if (l >= ql && r <= qr)
			return t[v];
		int mid = l + (r - l) / 2;
		return Sum(v * 2, l, mid, ql, qr) + Sum(v * 2 + 1, mid + 1, r, ql, qr);
	}
};

int n;

int cntHave = 0;

SegmentTree st[10];

map<int, int> index_;

void Inc(int lvl, int pos) {
	if (!index_.count(lvl)) {
		st[cntHave].Init(n);
		index_[lvl] = cntHave ++;
	}
	st[index_[lvl]].Add(1, 0, n - 1, pos);
}

int Get(int lvl, int l, int r) {
	if (!index_.count(lvl)) return 0;
	return st[index_[lvl]].Sum(1, 0, n - 1, l, r);
}

int find_best(int nn) {
	n = nn;
	int L = -1;
	while (L < n - 1) {
		int lo = L + 1, hi = n - 1;
		while (lo < hi) {
			int mid = lo + (hi - lo) / 2;
			Q ++;
			if (Q == 10000) {
				Q /= 0;
			}
			vector<int> vec = ask(mid);
			if (vec[0] + vec[1] == 0)
				return mid;
			int lvl = vec[0] + vec[1];
			int comp = 0;
			for (auto [a, b] : index_) if (a < lvl) comp += Get(a, 0, mid - 1);
			Inc(lvl, mid);
			if (vec[0] > comp)
				hi = mid - 1;
			else {
				lo = mid + 1;
			}
		}
		// cout << "lo = " << lo << '\n';
		Q ++;
		if (Q == 10000) {
			Q /= 0;
		}
		vector<int> vec = ask(lo);
		if (vec[0] + vec[1] == 0)
			return lo;
		int lvl = vec[0] + vec[1];
		Inc(lvl, lo);
		L = lo;
	}
	return n - 1;
}

/*
8
3 2 3 1 3 3 2 3
*/

Compilation message (stderr)

prize.cpp: In function 'int find_best(int)':
prize.cpp:66:7: warning: division by zero [-Wdiv-by-zero]
   66 |     Q /= 0;
      |     ~~^~~~
prize.cpp:84:6: warning: division by zero [-Wdiv-by-zero]
   84 |    Q /= 0;
      |    ~~^~~~

• Subtask 1: 20 / 20
• Subtask 2: 0 / 80