Submission #541819

#	Time	Username	Problem	Language	Result	Execution time	Memory
541819		keta_tsimakuridze	Cake 3 (JOI19_cake3)	C++14	0 / 100	1 ms	340 KiB

cake3

#include<bits/stdc++.h>
using namespace std;
 
#define ll long long
#define pii pair<ll,int>
#define f first
#define s second
#define endl "\n"
 
const int N = 2e5 + 5, mod = 1e9 + 7; //!
int t, n, val[N], a[N], cur = 0, ch1[2 * N], ch2[2 * N], c[4 * N];
ll s[4 * N]; 
pair<int,int> p[N];
ll ans;
int m;
void add(int u, int l, int r, int id, int t) {
	if(l > id || r < id) return;
	if(l == r) {
		c[u] += t;
		s[u] += t * val[id];
		return;
	}
	int mid = (l + r)/2;
	if(id <= mid) add(ch1[u], l, mid, id, t); 
	else add(ch2[u], mid + 1, r, id, t);
	c[u] = c[ch1[u]] + c[ch2[u]];
	s[u] = s[ch1[u]] + s[ch2[u]];
}
void add(int id, int t) {
	add(1, 1, cur, id, t);
}
ll get(int u,int l,int r,int k) {
	if(!k) return 0;
	if(l == r) {
		k = 0;
		return (ll)min(c[u], k) * val[l];
	}
	int mid = (l + r)/2;
	if(c[ch2[u]] <= k) return s[ch2[u]] + get(ch1[u], l, mid, k - c[ch2[u]]);
	return get(ch2[u], mid + 1, r, k);
}
void solve(int l,int r,int ql, int qr) {
	if(r < l || ql > qr) return;
	int mid = (l + r) / 2;
	pii opt; opt = {-2e17, ql};
	for(int i = mid; i >= max(l, qr + 1); i--) add(a[i], 1);
	if(mid < max(l, qr + 1))add(a[mid], 1);
	for(int i = min(qr, mid - 1);  i >= ql; i--) {
		add(a[i], 1); 
		if(mid - i + 1 >= m) 
		opt = max(opt, make_pair(get(1, 1, cur, m) - 2 * (p[mid].f - p[i].f), i));
	}
	ans = max(ans, opt.f);
 
	if(mid < max(l, qr + 1))add(a[mid], -1);
	for(int i = mid; i >= max(l, qr + 1); i--) add(a[i], -1);
	for(int i = min(qr, mid - 1);  i >= ql; i--) add(a[i], -1);
	if(l == r) return;
	for(int i = opt.s + 1; i <= min(qr, l - 1); i++) add(a[i], 1);
	solve(l, mid - 1, ql, opt.s); 
	for(int i = opt.s + 1; i <= min(qr, l - 1); i++) add(a[i], -1);
	for(int i = max(l, qr + 1); i <= mid; i++) add(a[i], 1);
	solve(mid + 1,r, opt.s, qr);
	for(int i = max(l, qr + 1); i <= mid; i++) add(a[i], -1);
	
}
main() {
	ios_base::sync_with_stdio(false),
	cin.tie(0),cout.tie(0);
	cin >> n >> m;
	ans = -1e18;
	vector<pair<int,int> > x;
	for(int i = 1; i <= 2 * n; i++) ch1[i] = 2 * i, ch2[i] = ch1[i] + 1;
	for(int i = 1; i <= n; i++) {
		cin >> p[i].s >> p[i].f;
	}
	sort(p + 1, p + n + 1);
	for(int i = 1; i <= n; i++) {
		x.push_back({p[i].s, i});
	}
	sort(x.begin(), x.end());
	for(int i = 0; i < x.size(); i++) {
		if(!i || x[i].f != x[i - 1].f) cur++;
		val[cur] = x[i].f;
		a[x[i].s] = cur;
	}
	solve(1, n, 1, n);
	cout << ans;
}

Compilation message (stderr)

cake3.cpp:67:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   67 | main() {
      | ^~~~
cake3.cpp: In function 'int main()':
cake3.cpp:82:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   82 |  for(int i = 0; i < x.size(); i++) {
      |                 ~~^~~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 19
• Subtask 3: 0 / 76