oj.uz

Submission #813387

# Submission time Handle Problem Language Result Execution time Memory
813387 2023-08-07T16:40:30 Z enerelt14 Diversity (CEOI21_diversity) C++14
64 / 100
 7000 ms 33104 KB 
#include <algorithm>
#include <array>
#include <bitset>
#include <cassert>
#include <chrono>
#include <climits>
#include <cmath>
#include <complex>
#include <cstring>
#include <functional>
#include <iomanip>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <vector>
using namespace std;

#define ff first
#define ss second
#define ll long long

const int MX = 3e5 + 5;
const int MX1 = 5e4 + 5;
const int B = 1341;

int N, Q;
int a[MX], num[MX], l[MX], r[MX];
ll ans[MX1];
pair<pair<int, int>, pair<int, int> > q[MX1];
ll val;

struct T{
	ll sum, left_sum, right_sum;
	T() {
		sum = left_sum = right_sum = 0;
	}
} tree[4 * MX];

void show(T x) {
	cout << x.left_sum << " " << x.right_sum << " " << x.sum << "\n";
}

void update(int id, int l, int r, int pos, int x) {
	if(l > pos || pos > r) return;
	tree[id].sum += x;
	tree[id].left_sum += x * (pos - l + 1);
	tree[id].right_sum += x * (r - pos + 1);
	if(l == r) return;
	int mid = (l + r) / 2;
	update(id * 2 + 1, l, mid, pos, x);
	update(id * 2 + 2, mid + 1, r, pos, x);
}

T query(int id, int l, int r, int L, int R) {
	T res;
	if(L > r || l > R) return res;
	if(L <= l && r <= R) return tree[id];
	int mid = (l + r) / 2;
	if(L > mid) return query(id * 2 + 2, mid + 1, r, L, R);
	if(R <= mid) return query(id * 2 + 1, l, mid, L, R);
	T x = query(id * 2 + 1, l, mid, L, R), y = query(id * 2 + 2, mid + 1, r, L, R);
	res.sum = x.sum + y.sum;
	res.left_sum = x.left_sum + y.left_sum + y.sum * (mid - L + 1);
	res.right_sum = x.right_sum + y.right_sum + x.sum * (R - mid);
	return res;
}

int prv(int x) {
	if(x <= (N - 1) / 2) return N - x;
	else return N - x - 1;
}

int nxt(int x) {
	if(x <= N / 2) return N - x - 1;
	else return N - x;
}

void add(int x) {
	num[x]++;
	l[num[x]] = r[num[x] - 1];
	if(r[num[x]] == -1) r[num[x]] = l[num[x]];
	if(r[num[x] - 1] == l[num[x] - 1]) r[num[x] - 1] = l[num[x] - 1] = -1;
	else r[num[x] - 1] = prv(r[num[x] - 1]);
	update(0, 0, N - 1, l[num[x]], 1);
	T y, z;
	if(l[num[x]] != 0) y = query(0, 0, N - 1, 0, l[num[x]] - 1);
	if(l[num[x]] != N - 1) z = query(0, 0, N - 1, l[num[x]] + 1, N - 1);
	val += y.right_sum + y.sum + z.left_sum + z.sum;
	val += num[x];
	// show(y);
	// show(z);
	// cout << val << "\n---------\n";
}

void remove(int x) {
	num[x]--;
	r[num[x]] = l[num[x] + 1];
	if(l[num[x]] == -1) l[num[x]] = r[num[x]];
	if(r[num[x] + 1] == l[num[x] + 1]) r[num[x] + 1] = l[num[x] + 1] = -1;
	else l[num[x] + 1] = nxt(l[num[x] + 1]);
	update(0, 0, N - 1, r[num[x]], -1);
	T y, z;
	if(r[num[x]] != 0) y = query(0, 0, N - 1, 0, r[num[x]] - 1);
	if(r[num[x]] != N - 1) z = query(0, 0, N - 1, r[num[x]] + 1, N - 1);
	val -= y.right_sum + y.sum + z.left_sum + z.sum;
	val -= (num[x] + 1);
}

int main() {
	memset(l, -1, sizeof(l));
	memset(r, -1, sizeof(r));
	cin >> N >> Q;
	l[0] = 0;
	r[0] = N / 2;
	for(int i = 0; i < N; i++) cin >> a[i];
	for(int i = 0; i < Q; i++) {
		int l, r;
		cin >> l >> r;
		l--;
		r--;
		q[i] = {{l / B, r}, {l, i}};
	}
	sort(q, q + Q);
	for(int i = 0; i < Q; i++) {
		if(i == 0) {
			for(int j = 0; j <= q[i].ff.ss; j++) add(a[j]);
			for(int j = 0; j < q[i].ss.ff; j++) remove(a[j]);
			ans[q[i].ss.ss] = val;
			continue;
		}
		for(int j = q[i - 1].ff.ss + 1; j <= q[i].ff.ss; j++) add(a[j]);
		for(int j = q[i - 1].ff.ss; j > q[i].ff.ss; j--) remove(a[j]);
		for(int j = q[i - 1].ss.ff; j < q[i].ss.ff; j++) remove(a[j]);
		for(int j = q[i - 1].ss.ff - 1; j >= q[i].ss.ff; j--) add(a[j]);
		ans[q[i].ss.ss] = val;
	}
	for(int i = 0; i < Q; i++) {
		cout << ans[i] << "\n";
	}
}

Subtask #1
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 30840 KB Output is correct
2 Correct 16 ms 30804 KB Output is correct
3 Correct 14 ms 30804 KB Output is correct
4 Correct 14 ms 30848 KB Output is correct
5 Correct 14 ms 30812 KB Output is correct
6 Correct 14 ms 30792 KB Output is correct
7 Correct 13 ms 30804 KB Output is correct
8 Correct 14 ms 30844 KB Output is correct
9 Correct 13 ms 30804 KB Output is correct
10 Correct 13 ms 30792 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 30848 KB Output is correct
2 Correct 14 ms 30848 KB Output is correct
3 Correct 20 ms 30848 KB Output is correct
4 Correct 71 ms 31376 KB Output is correct
5 Correct 95 ms 31988 KB Output is correct
6 Correct 134 ms 32380 KB Output is correct
7 Correct 129 ms 32376 KB Output is correct
8 Correct 143 ms 32380 KB Output is correct
9 Correct 133 ms 32368 KB Output is correct
10 Correct 131 ms 32384 KB Output is correct
11 Correct 136 ms 32376 KB Output is correct
12 Correct 137 ms 32460 KB Output is correct
13 Correct 177 ms 32380 KB Output is correct
14 Correct 195 ms 32380 KB Output is correct
15 Correct 207 ms 32376 KB Output is correct

Subtask #3
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 30848 KB Output is correct
2 Correct 14 ms 30848 KB Output is correct
3 Correct 20 ms 30848 KB Output is correct
4 Correct 71 ms 31376 KB Output is correct
5 Correct 95 ms 31988 KB Output is correct
6 Correct 134 ms 32380 KB Output is correct
7 Correct 129 ms 32376 KB Output is correct
8 Correct 143 ms 32380 KB Output is correct
9 Correct 133 ms 32368 KB Output is correct
10 Correct 131 ms 32384 KB Output is correct
11 Correct 136 ms 32376 KB Output is correct
12 Correct 137 ms 32460 KB Output is correct
13 Correct 177 ms 32380 KB Output is correct
14 Correct 195 ms 32380 KB Output is correct
15 Correct 207 ms 32376 KB Output is correct
16 Correct 12 ms 30844 KB Output is correct
17 Correct 11 ms 30804 KB Output is correct
18 Correct 14 ms 30908 KB Output is correct
19 Correct 50 ms 31460 KB Output is correct
20 Correct 95 ms 31924 KB Output is correct
21 Correct 133 ms 32360 KB Output is correct
22 Correct 134 ms 32376 KB Output is correct
23 Correct 133 ms 32332 KB Output is correct
24 Correct 133 ms 32380 KB Output is correct
25 Correct 138 ms 32376 KB Output is correct
26 Correct 139 ms 32332 KB Output is correct
27 Correct 158 ms 32332 KB Output is correct
28 Correct 141 ms 32384 KB Output is correct
29 Correct 151 ms 32380 KB Output is correct
30 Correct 144 ms 32380 KB Output is correct

Subtask #4
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 30848 KB Output is correct
2 Correct 14 ms 30848 KB Output is correct
3 Correct 20 ms 30848 KB Output is correct
4 Correct 71 ms 31376 KB Output is correct
5 Correct 95 ms 31988 KB Output is correct
6 Correct 134 ms 32380 KB Output is correct
7 Correct 129 ms 32376 KB Output is correct
8 Correct 143 ms 32380 KB Output is correct
9 Correct 133 ms 32368 KB Output is correct
10 Correct 131 ms 32384 KB Output is correct
11 Correct 136 ms 32376 KB Output is correct
12 Correct 137 ms 32460 KB Output is correct
13 Correct 177 ms 32380 KB Output is correct
14 Correct 195 ms 32380 KB Output is correct
15 Correct 207 ms 32376 KB Output is correct
16 Correct 12 ms 30844 KB Output is correct
17 Correct 11 ms 30804 KB Output is correct
18 Correct 14 ms 30908 KB Output is correct
19 Correct 50 ms 31460 KB Output is correct
20 Correct 95 ms 31924 KB Output is correct
21 Correct 133 ms 32360 KB Output is correct
22 Correct 134 ms 32376 KB Output is correct
23 Correct 133 ms 32332 KB Output is correct
24 Correct 133 ms 32380 KB Output is correct
25 Correct 138 ms 32376 KB Output is correct
26 Correct 139 ms 32332 KB Output is correct
27 Correct 158 ms 32332 KB Output is correct
28 Correct 141 ms 32384 KB Output is correct
29 Correct 151 ms 32380 KB Output is correct
30 Correct 144 ms 32380 KB Output is correct
31 Correct 14 ms 30816 KB Output is correct
32 Correct 15 ms 30840 KB Output is correct
33 Correct 13 ms 30804 KB Output is correct
34 Correct 17 ms 30812 KB Output is correct
35 Correct 17 ms 30852 KB Output is correct
36 Correct 17 ms 30928 KB Output is correct
37 Correct 37 ms 31208 KB Output is correct
38 Correct 36 ms 31112 KB Output is correct
39 Correct 63 ms 31596 KB Output is correct
40 Correct 117 ms 31880 KB Output is correct
41 Correct 167 ms 32388 KB Output is correct
42 Correct 166 ms 32332 KB Output is correct
43 Correct 170 ms 32372 KB Output is correct
44 Correct 178 ms 32384 KB Output is correct
45 Correct 170 ms 32340 KB Output is correct
46 Correct 172 ms 32384 KB Output is correct
47 Correct 172 ms 32384 KB Output is correct
48 Correct 186 ms 32384 KB Output is correct
49 Correct 170 ms 32392 KB Output is correct
50 Correct 169 ms 32384 KB Output is correct
51 Correct 173 ms 32444 KB Output is correct
52 Correct 181 ms 32320 KB Output is correct
53 Correct 168 ms 32348 KB Output is correct
54 Correct 167 ms 32388 KB Output is correct
55 Correct 177 ms 32384 KB Output is correct
56 Correct 167 ms 32388 KB Output is correct
57 Correct 167 ms 32384 KB Output is correct
58 Correct 183 ms 32436 KB Output is correct
59 Correct 206 ms 32388 KB Output is correct
60 Correct 170 ms 32388 KB Output is correct

Subtask #5
26.0 / 26.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 30840 KB Output is correct
2 Correct 16 ms 30804 KB Output is correct
3 Correct 14 ms 30804 KB Output is correct
4 Correct 14 ms 30848 KB Output is correct
5 Correct 14 ms 30812 KB Output is correct
6 Correct 14 ms 30792 KB Output is correct
7 Correct 13 ms 30804 KB Output is correct
8 Correct 14 ms 30844 KB Output is correct
9 Correct 13 ms 30804 KB Output is correct
10 Correct 13 ms 30792 KB Output is correct
11 Correct 13 ms 30848 KB Output is correct
12 Correct 14 ms 30848 KB Output is correct
13 Correct 20 ms 30848 KB Output is correct
14 Correct 71 ms 31376 KB Output is correct
15 Correct 95 ms 31988 KB Output is correct
16 Correct 134 ms 32380 KB Output is correct
17 Correct 129 ms 32376 KB Output is correct
18 Correct 143 ms 32380 KB Output is correct
19 Correct 133 ms 32368 KB Output is correct
20 Correct 131 ms 32384 KB Output is correct
21 Correct 136 ms 32376 KB Output is correct
22 Correct 137 ms 32460 KB Output is correct
23 Correct 177 ms 32380 KB Output is correct
24 Correct 195 ms 32380 KB Output is correct
25 Correct 207 ms 32376 KB Output is correct
26 Correct 12 ms 30844 KB Output is correct
27 Correct 11 ms 30804 KB Output is correct
28 Correct 14 ms 30908 KB Output is correct
29 Correct 50 ms 31460 KB Output is correct
30 Correct 95 ms 31924 KB Output is correct
31 Correct 133 ms 32360 KB Output is correct
32 Correct 134 ms 32376 KB Output is correct
33 Correct 133 ms 32332 KB Output is correct
34 Correct 133 ms 32380 KB Output is correct
35 Correct 138 ms 32376 KB Output is correct
36 Correct 139 ms 32332 KB Output is correct
37 Correct 158 ms 32332 KB Output is correct
38 Correct 141 ms 32384 KB Output is correct
39 Correct 151 ms 32380 KB Output is correct
40 Correct 144 ms 32380 KB Output is correct
41 Correct 14 ms 30816 KB Output is correct
42 Correct 15 ms 30840 KB Output is correct
43 Correct 13 ms 30804 KB Output is correct
44 Correct 17 ms 30812 KB Output is correct
45 Correct 17 ms 30852 KB Output is correct
46 Correct 17 ms 30928 KB Output is correct
47 Correct 37 ms 31208 KB Output is correct
48 Correct 36 ms 31112 KB Output is correct
49 Correct 63 ms 31596 KB Output is correct
50 Correct 117 ms 31880 KB Output is correct
51 Correct 167 ms 32388 KB Output is correct
52 Correct 166 ms 32332 KB Output is correct
53 Correct 170 ms 32372 KB Output is correct
54 Correct 178 ms 32384 KB Output is correct
55 Correct 170 ms 32340 KB Output is correct
56 Correct 172 ms 32384 KB Output is correct
57 Correct 172 ms 32384 KB Output is correct
58 Correct 186 ms 32384 KB Output is correct
59 Correct 170 ms 32392 KB Output is correct
60 Correct 169 ms 32384 KB Output is correct
61 Correct 173 ms 32444 KB Output is correct
62 Correct 181 ms 32320 KB Output is correct
63 Correct 168 ms 32348 KB Output is correct
64 Correct 167 ms 32388 KB Output is correct
65 Correct 177 ms 32384 KB Output is correct
66 Correct 167 ms 32388 KB Output is correct
67 Correct 167 ms 32384 KB Output is correct
68 Correct 183 ms 32436 KB Output is correct
69 Correct 206 ms 32388 KB Output is correct
70 Correct 170 ms 32388 KB Output is correct
71 Correct 41 ms 31252 KB Output is correct
72 Correct 38 ms 31264 KB Output is correct
73 Correct 39 ms 31260 KB Output is correct
74 Correct 39 ms 31236 KB Output is correct
75 Correct 48 ms 31216 KB Output is correct
76 Correct 67 ms 31572 KB Output is correct
77 Correct 91 ms 31620 KB Output is correct
78 Correct 92 ms 31624 KB Output is correct
79 Correct 71 ms 31624 KB Output is correct
80 Correct 70 ms 31628 KB Output is correct
81 Correct 125 ms 32032 KB Output is correct
82 Correct 128 ms 31928 KB Output is correct
83 Correct 127 ms 31964 KB Output is correct
84 Correct 124 ms 31948 KB Output is correct
85 Correct 177 ms 31708 KB Output is correct
86 Correct 127 ms 31784 KB Output is correct
87 Correct 129 ms 31808 KB Output is correct
88 Correct 129 ms 31820 KB Output is correct
89 Correct 144 ms 31744 KB Output is correct
90 Correct 132 ms 31820 KB Output is correct
91 Correct 191 ms 32504 KB Output is correct
92 Correct 204 ms 32460 KB Output is correct
93 Correct 226 ms 32508 KB Output is correct
94 Correct 203 ms 32428 KB Output is correct
95 Correct 275 ms 32516 KB Output is correct
96 Correct 200 ms 32852 KB Output is correct
97 Correct 212 ms 32860 KB Output is correct
98 Correct 250 ms 32900 KB Output is correct
99 Correct 273 ms 32916 KB Output is correct
100 Correct 209 ms 32908 KB Output is correct
101 Correct 198 ms 32912 KB Output is correct
102 Correct 216 ms 32904 KB Output is correct
103 Correct 212 ms 32904 KB Output is correct
104 Correct 217 ms 32908 KB Output is correct
105 Correct 197 ms 32912 KB Output is correct
106 Correct 192 ms 33072 KB Output is correct
107 Correct 209 ms 33104 KB Output is correct
108 Correct 191 ms 33096 KB Output is correct
109 Correct 193 ms 33096 KB Output is correct
110 Correct 214 ms 33080 KB Output is correct
111 Correct 194 ms 33096 KB Output is correct
112 Correct 193 ms 33100 KB Output is correct
113 Correct 193 ms 33096 KB Output is correct
114 Correct 191 ms 33008 KB Output is correct
115 Correct 191 ms 33100 KB Output is correct

Subtask #6
0 / 36.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 30840 KB Output is correct
2 Correct 16 ms 30804 KB Output is correct
3 Correct 14 ms 30804 KB Output is correct
4 Correct 14 ms 30848 KB Output is correct
5 Correct 14 ms 30812 KB Output is correct
6 Correct 14 ms 30792 KB Output is correct
7 Correct 13 ms 30804 KB Output is correct
8 Correct 14 ms 30844 KB Output is correct
9 Correct 13 ms 30804 KB Output is correct
10 Correct 13 ms 30792 KB Output is correct
11 Correct 13 ms 30848 KB Output is correct
12 Correct 14 ms 30848 KB Output is correct
13 Correct 20 ms 30848 KB Output is correct
14 Correct 71 ms 31376 KB Output is correct
15 Correct 95 ms 31988 KB Output is correct
16 Correct 134 ms 32380 KB Output is correct
17 Correct 129 ms 32376 KB Output is correct
18 Correct 143 ms 32380 KB Output is correct
19 Correct 133 ms 32368 KB Output is correct
20 Correct 131 ms 32384 KB Output is correct
21 Correct 136 ms 32376 KB Output is correct
22 Correct 137 ms 32460 KB Output is correct
23 Correct 177 ms 32380 KB Output is correct
24 Correct 195 ms 32380 KB Output is correct
25 Correct 207 ms 32376 KB Output is correct
26 Correct 12 ms 30844 KB Output is correct
27 Correct 11 ms 30804 KB Output is correct
28 Correct 14 ms 30908 KB Output is correct
29 Correct 50 ms 31460 KB Output is correct
30 Correct 95 ms 31924 KB Output is correct
31 Correct 133 ms 32360 KB Output is correct
32 Correct 134 ms 32376 KB Output is correct
33 Correct 133 ms 32332 KB Output is correct
34 Correct 133 ms 32380 KB Output is correct
35 Correct 138 ms 32376 KB Output is correct
36 Correct 139 ms 32332 KB Output is correct
37 Correct 158 ms 32332 KB Output is correct
38 Correct 141 ms 32384 KB Output is correct
39 Correct 151 ms 32380 KB Output is correct
40 Correct 144 ms 32380 KB Output is correct
41 Correct 14 ms 30816 KB Output is correct
42 Correct 15 ms 30840 KB Output is correct
43 Correct 13 ms 30804 KB Output is correct
44 Correct 17 ms 30812 KB Output is correct
45 Correct 17 ms 30852 KB Output is correct
46 Correct 17 ms 30928 KB Output is correct
47 Correct 37 ms 31208 KB Output is correct
48 Correct 36 ms 31112 KB Output is correct
49 Correct 63 ms 31596 KB Output is correct
50 Correct 117 ms 31880 KB Output is correct
51 Correct 167 ms 32388 KB Output is correct
52 Correct 166 ms 32332 KB Output is correct
53 Correct 170 ms 32372 KB Output is correct
54 Correct 178 ms 32384 KB Output is correct
55 Correct 170 ms 32340 KB Output is correct
56 Correct 172 ms 32384 KB Output is correct
57 Correct 172 ms 32384 KB Output is correct
58 Correct 186 ms 32384 KB Output is correct
59 Correct 170 ms 32392 KB Output is correct
60 Correct 169 ms 32384 KB Output is correct
61 Correct 173 ms 32444 KB Output is correct
62 Correct 181 ms 32320 KB Output is correct
63 Correct 168 ms 32348 KB Output is correct
64 Correct 167 ms 32388 KB Output is correct
65 Correct 177 ms 32384 KB Output is correct
66 Correct 167 ms 32388 KB Output is correct
67 Correct 167 ms 32384 KB Output is correct
68 Correct 183 ms 32436 KB Output is correct
69 Correct 206 ms 32388 KB Output is correct
70 Correct 170 ms 32388 KB Output is correct
71 Correct 41 ms 31252 KB Output is correct
72 Correct 38 ms 31264 KB Output is correct
73 Correct 39 ms 31260 KB Output is correct
74 Correct 39 ms 31236 KB Output is correct
75 Correct 48 ms 31216 KB Output is correct
76 Correct 67 ms 31572 KB Output is correct
77 Correct 91 ms 31620 KB Output is correct
78 Correct 92 ms 31624 KB Output is correct
79 Correct 71 ms 31624 KB Output is correct
80 Correct 70 ms 31628 KB Output is correct
81 Correct 125 ms 32032 KB Output is correct
82 Correct 128 ms 31928 KB Output is correct
83 Correct 127 ms 31964 KB Output is correct
84 Correct 124 ms 31948 KB Output is correct
85 Correct 177 ms 31708 KB Output is correct
86 Correct 127 ms 31784 KB Output is correct
87 Correct 129 ms 31808 KB Output is correct
88 Correct 129 ms 31820 KB Output is correct
89 Correct 144 ms 31744 KB Output is correct
90 Correct 132 ms 31820 KB Output is correct
91 Correct 191 ms 32504 KB Output is correct
92 Correct 204 ms 32460 KB Output is correct
93 Correct 226 ms 32508 KB Output is correct
94 Correct 203 ms 32428 KB Output is correct
95 Correct 275 ms 32516 KB Output is correct
96 Correct 200 ms 32852 KB Output is correct
97 Correct 212 ms 32860 KB Output is correct
98 Correct 250 ms 32900 KB Output is correct
99 Correct 273 ms 32916 KB Output is correct
100 Correct 209 ms 32908 KB Output is correct
101 Correct 198 ms 32912 KB Output is correct
102 Correct 216 ms 32904 KB Output is correct
103 Correct 212 ms 32904 KB Output is correct
104 Correct 217 ms 32908 KB Output is correct
105 Correct 197 ms 32912 KB Output is correct
106 Correct 192 ms 33072 KB Output is correct
107 Correct 209 ms 33104 KB Output is correct
108 Correct 191 ms 33096 KB Output is correct
109 Correct 193 ms 33096 KB Output is correct
110 Correct 214 ms 33080 KB Output is correct
111 Correct 194 ms 33096 KB Output is correct
112 Correct 193 ms 33100 KB Output is correct
113 Correct 193 ms 33096 KB Output is correct
114 Correct 191 ms 33008 KB Output is correct
115 Correct 191 ms 33100 KB Output is correct
116 Correct 4605 ms 32544 KB Output is correct
117 Correct 4942 ms 32548 KB Output is correct
118 Correct 6645 ms 32824 KB Output is correct
119 Execution timed out 7083 ms 32296 KB Time limit exceeded
120 Halted 0 ms 0 KB -