Submission #1111655

#	Time	Username	Problem	Language	Result	Execution time	Memory
1111655		Onur_Ilgaz	New Home (APIO18_new_home)	C++17	10 / 100	208 ms	35496 KiB

new_home

#include <bits/stdc++.h>
#define fast cin.tie(0)->sync_with_stdio(0);
#define int long long
#define inf ((int)1e18)
using namespace std;

int32_t main(){
	fast
	int n, k, q;
	cin >> n >> k >> q;
	vector <pair<int, int> > arr, ranges;
	for(int i = 0; i < n; i++) {
		int x, t, a, b;
		cin >> x >> t >> a >> b;
		arr.push_back({x, t});
	}
	sort(arr.begin(), arr.end());
	int l = 0, r = -1, cnt = 0;
	vector <int> have(k + 1);
	while(r < n - 1) {
		r++;
		auto [pos, type] = arr[r];
		if(!have[type]) cnt++;
		have[type]++;
		while(cnt == k) {
			auto [tpos, ttype] = arr[l];
			have[ttype]--;
			if(!have[ttype]) cnt--;
			// add to ranges
			ranges.push_back({tpos, pos});
			l++;
		}
	}
	// for(auto it:ranges) {
	// 	cout << it.first << ' ' << it.second << '\n';
	// }
	if(ranges.empty()) {
		for(int i = 0; i < q; i++) {
			cout << -1 << "\n";
		}
		return 0;
	}
	vector <pair<int, int> > qq;
	for(int i = 0; i < q; i++) {
		int x, t;
		cin >> x >> t;
		qq.push_back({x, i});
	}
	sort(qq.begin(), qq.end());
	vector <int> ans(q);
	auto calc = [&](int rind, int pos) {
		if(rind == ranges.size()) return inf;
		auto [l, r] = ranges[rind];
		return max(abs(pos - l), abs(pos - r));
	};
	int now = 0;
	for(auto [pos, ans_ind] : qq) {
		while(calc(now + 1, pos) <= calc(now, pos)) {
			now++;
		}
		ans[ans_ind] = calc(now, pos);
	}
	for(auto it:ans) {
		cout << it << "\n";
	}
}

Compilation message (stderr)

new_home.cpp: In lambda function:
new_home.cpp:52:11: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   52 |   if(rind == ranges.size()) return inf;
      |      ~~~~~^~~~~~~~~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 7
• Subtask 3: 10 / 10
• Subtask 4: 0 / 23
• Subtask 5: 0 / 35
• Subtask 6: 0 / 20