Submission #207734

# Submission time Handle Problem Language Result Execution time Memory
207734 2020-03-08T17:28:54 Z cgiosy Dancing Elephants (IOI11_elephants) C++17
100 / 100
3801 ms 7560 KB
#include <bits/stdc++.h>
#define lb(a, x) lower_bound(begin(a), end(a), x)
using namespace std;
constexpr int MX=150000, B=400, C=(MX+B-1)/B;

struct iii { int x, n, y; };
bool operator<(const iii& a, int b) { return a.x<b; }
bool operator<(int a, const iii& b) { return a<b.x; }
int A[MX], X[MX], N, M, Q;
vector<iii> D[C];
void calc_bucket(vector<iii>& v) {
	for(int n=v.size(), i=n, p=n; i--;) {
		while(p && v[i].x<v[p-1].x-M) p--;
		if(p==n) v[i].n=1, v[i].y=v[i].x+M;
		else v[i].n=v[p].n+1, v[i].y=v[p].y;
	}
}
void init_buckets() {
	int k=0;
	for(auto&v:D) {
		for(auto[x,n,y]:v) X[k++]=x;
		v.clear();
	}
	for(int i=0, j=0; i<N;) {
		D[j].push_back({X[i], 0, 0});
		if(++i%B==0 || i==N) calc_bucket(D[j++]);
	}
}
void del(int x) {
	for(auto&v:D) if(v.size() && v.front().x<=x && x<=v.back().x) {
		v.erase(lb(v, x));
		calc_bucket(v);
		break;
	}
}
void add(int x) {
	int p=0, n=N-1;
	for(auto&v:D) if(!(n-=v.size()) || v.size() && p<=x && x<=(p=v.back().x)) {
		v.insert(lb(v, x), {x, 0, 0});
		calc_bucket(v);
		break;
	}
}
int answer() {
	int p=0, n=0;
	for(auto&v:D) if(v.size() && p<=v.back().x) {
		auto[x,m,y]=*lb(v, p);
		n+=m, p=y+1;
	}
	return n;
}

void init(int n, int m, int E[]) {
	N=n, M=m;
	for(int i=0; i<C; i++) D[i].reserve(B);
	for(int i=0; i<N; i++) D[i/B].push_back({A[i]=E[i], 0, 0});
}
int update(int n, int x) {
	if(Q++%B==0) init_buckets();
	del(A[n]);
	add(A[n]=x);
	return answer();
}

Compilation message

elephants.cpp: In function 'void init_buckets()':
elephants.cpp:21:17: warning: unused variable 'n' [-Wunused-variable]
   for(auto[x,n,y]:v) X[k++]=x;
                 ^
elephants.cpp:21:17: warning: unused variable 'y' [-Wunused-variable]
elephants.cpp: In function 'void add(int)':
elephants.cpp:38:54: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
  for(auto&v:D) if(!(n-=v.size()) || v.size() && p<=x && x<=(p=v.back().x)) {
                                     ~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~
elephants.cpp: In function 'int answer()':
elephants.cpp:47:13: warning: unused variable 'x' [-Wunused-variable]
   auto[x,m,y]=*lb(v, p);
             ^
# Verdict Execution time Memory Grader output
1 Correct 6 ms 1912 KB Output is correct
2 Correct 5 ms 1912 KB Output is correct
3 Correct 6 ms 1912 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 6 ms 1912 KB Output is correct
2 Correct 5 ms 1912 KB Output is correct
3 Correct 6 ms 1912 KB Output is correct
4 Correct 5 ms 1912 KB Output is correct
5 Correct 6 ms 1912 KB Output is correct
6 Correct 5 ms 1912 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 6 ms 1912 KB Output is correct
2 Correct 5 ms 1912 KB Output is correct
3 Correct 6 ms 1912 KB Output is correct
4 Correct 5 ms 1912 KB Output is correct
5 Correct 6 ms 1912 KB Output is correct
6 Correct 5 ms 1912 KB Output is correct
7 Correct 375 ms 2812 KB Output is correct
8 Correct 416 ms 2936 KB Output is correct
9 Correct 408 ms 3448 KB Output is correct
10 Correct 421 ms 3064 KB Output is correct
11 Correct 386 ms 3064 KB Output is correct
12 Correct 655 ms 3704 KB Output is correct
13 Correct 435 ms 3192 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 6 ms 1912 KB Output is correct
2 Correct 5 ms 1912 KB Output is correct
3 Correct 6 ms 1912 KB Output is correct
4 Correct 5 ms 1912 KB Output is correct
5 Correct 6 ms 1912 KB Output is correct
6 Correct 5 ms 1912 KB Output is correct
7 Correct 375 ms 2812 KB Output is correct
8 Correct 416 ms 2936 KB Output is correct
9 Correct 408 ms 3448 KB Output is correct
10 Correct 421 ms 3064 KB Output is correct
11 Correct 386 ms 3064 KB Output is correct
12 Correct 655 ms 3704 KB Output is correct
13 Correct 435 ms 3192 KB Output is correct
14 Correct 402 ms 3192 KB Output is correct
15 Correct 629 ms 3360 KB Output is correct
16 Correct 1076 ms 3960 KB Output is correct
17 Correct 1123 ms 4728 KB Output is correct
18 Correct 1226 ms 4472 KB Output is correct
19 Correct 589 ms 3704 KB Output is correct
20 Correct 1114 ms 4600 KB Output is correct
21 Correct 1092 ms 4520 KB Output is correct
22 Correct 723 ms 3576 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 6 ms 1912 KB Output is correct
2 Correct 5 ms 1912 KB Output is correct
3 Correct 6 ms 1912 KB Output is correct
4 Correct 5 ms 1912 KB Output is correct
5 Correct 6 ms 1912 KB Output is correct
6 Correct 5 ms 1912 KB Output is correct
7 Correct 375 ms 2812 KB Output is correct
8 Correct 416 ms 2936 KB Output is correct
9 Correct 408 ms 3448 KB Output is correct
10 Correct 421 ms 3064 KB Output is correct
11 Correct 386 ms 3064 KB Output is correct
12 Correct 655 ms 3704 KB Output is correct
13 Correct 435 ms 3192 KB Output is correct
14 Correct 402 ms 3192 KB Output is correct
15 Correct 629 ms 3360 KB Output is correct
16 Correct 1076 ms 3960 KB Output is correct
17 Correct 1123 ms 4728 KB Output is correct
18 Correct 1226 ms 4472 KB Output is correct
19 Correct 589 ms 3704 KB Output is correct
20 Correct 1114 ms 4600 KB Output is correct
21 Correct 1092 ms 4520 KB Output is correct
22 Correct 723 ms 3576 KB Output is correct
23 Correct 3028 ms 6880 KB Output is correct
24 Correct 3329 ms 7280 KB Output is correct
25 Correct 2290 ms 6672 KB Output is correct
26 Correct 2917 ms 5752 KB Output is correct
27 Correct 2800 ms 5696 KB Output is correct
28 Correct 1352 ms 3704 KB Output is correct
29 Correct 1306 ms 3788 KB Output is correct
30 Correct 1363 ms 3704 KB Output is correct
31 Correct 1310 ms 3832 KB Output is correct
32 Correct 2213 ms 5752 KB Output is correct
33 Correct 1945 ms 5752 KB Output is correct
34 Correct 2384 ms 5700 KB Output is correct
35 Correct 1806 ms 7512 KB Output is correct
36 Correct 1613 ms 5752 KB Output is correct
37 Correct 2916 ms 7560 KB Output is correct
38 Correct 2514 ms 5692 KB Output is correct
39 Correct 2084 ms 5628 KB Output is correct
40 Correct 2755 ms 5624 KB Output is correct
41 Correct 3767 ms 7544 KB Output is correct
42 Correct 3801 ms 7416 KB Output is correct