Submission #519390

# Submission time Handle Problem Language Result Execution time Memory
519390 2022-01-26T10:02:55 Z Keshi Diversity (CEOI21_diversity) C++17
64 / 100
7000 ms 46332 KB
//In the name of God
#include <bits/stdc++.h>
#pragma GCC optimize("O2")
#pragma GCC optimize("Ofast")
using namespace std;

typedef long long ll;
typedef pair<ll, ll> pll;

const ll maxn = 3e5 + 100;
const ll mod = 1e9 + 7;
const ll inf = 1e18;
const int sq = 512;

#define fast_io ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define file_io freopen("input.txt", "r+", stdin);freopen("output.txt", "w+", stdout);
#define pb push_back
#define Mp make_pair
#define F first
#define S second
#define Sz(x) ll((x).size())
#define all(x) (x).begin(), (x).end()
#define lc (id << 1)
#define rc (lc | 1)

ll pw(ll a, ll b){
	ll c = 1;
	while(b){
		if(b & 1) c = c * a % mod;
		a = a * a % mod;
		b >>= 1;
	}
	return c;
}

int n, q, a[maxn], c[maxn], b[maxn], N, ind[maxn], cnt[maxn], cn[maxn];

namespace A{
ll seg[maxn << 2], sg[maxn << 2], lz[maxn << 2];

inline void Do(int id, int s, int e, ll x){
	seg[id] += x * ((e - s) * (x + 1) + 2 * sg[id]) / 2;
	sg[id] += x * (e - s);
	lz[id] += x;
	return;
}

void upd(int id, int s, int e, int l, int r, ll x){
	if(r <= s || e <= l) return;
	if(l <= s && e <= r){
		Do(id, s, e, x);
		return;
	}
	int mid = (s + e) >> 1;
	Do(lc, s, mid, lz[id]);
	Do(rc, mid, e, lz[id]);
	lz[id] = 0;
	upd(lc, s, mid, l, r, x);
	upd(rc, mid, e, l, r, x);
	seg[id] = seg[lc] + seg[rc];
	sg[id] = sg[lc] + sg[rc];
	return;
}
};
namespace B{
ll seg[maxn << 2], sg[maxn << 2], lz[maxn << 2];

inline void Do(int id, int s, int e, ll x){
	seg[id] += x * ((e - s) * (x + 1) + 2 * sg[id]) / 2;
	sg[id] += x * (e - s);
	lz[id] += x;
	return;
}

void upd(int id, int s, int e, int l, int r, ll x){
	if(r <= s || e <= l) return;
	if(l <= s && e <= r){
		Do(id, s, e, x);
		return;
	}
	int mid = (s + e) >> 1;
	Do(lc, s, mid, lz[id]);
	Do(rc, mid, e, lz[id]);
	lz[id] = 0;
	upd(lc, s, mid, l, r, x);
	upd(rc, mid, e, l, r, x);
	seg[id] = seg[lc] + seg[rc];
	sg[id] = sg[lc] + sg[rc];
	return;
}
};

void upd(int i, int x){
	int j = ind[i];
	cnt[i] += x;
	A::upd(1, 0, N, j, N, x);
	B::upd(1, 0, N, N - 1 - j, N, x);
	return;
}
void add(int x){
	x = a[x];
	upd(upper_bound(cnt, cnt + N, cn[x]) - cnt - 1, 1);
	cn[x]++;
}
void rem(int x){
	x = a[x];
	upd(lower_bound(cnt, cnt + N, cn[x]) - cnt, -1);
	cn[x]--;
}

pll qq[maxn];
int p[maxn];
ll ans[maxn];

bool cmp(int i, int j){
	return Mp(qq[i].F / sq, qq[i].S) < Mp(qq[j].F / sq, qq[j].S);
}

int main(){
	fast_io;
	{
	N = maxn - 10;
	int l = 0, r = N - 1;
	for(int i = 0; i < N; i++){
		if(i & 1) ind[i] = r--;
		else ind[i] = l++;
	}
	}

	cin >> n >> q;
	for(int i = 0; i < n; i++){
		cin >> a[i];
	}
	for(int i = 0; i < q; i++){
		int l, r;
		cin >> l >> r;
		qq[i] = Mp(l, r);
		p[i] = i;
	}
	sort(p, p + q, cmp);
	int L = 0, R = 0;
	for(int i = 0; i < q; i++){
		int l, r;
		l = qq[p[i]].F;
		r = qq[p[i]].S;
		l--;
		while(R < r) add(R++);
		while(L > l) add(--L);
		while(R > r) rem(--R);
		while(L < l) rem(L++);
		ll m = r - l;
		ans[p[i]] = m * (m + 1) / 2 * (N + 2) - A::seg[1] - B::seg[1];
	}
	for(int i = 0; i < q; i++){
		cout << ans[i] << "\n";
	}
	return 0;
}
# Verdict Execution time Memory Grader output
1 Correct 1 ms 1740 KB Output is correct
2 Correct 1 ms 1868 KB Output is correct
3 Correct 2 ms 1868 KB Output is correct
4 Correct 1 ms 1868 KB Output is correct
5 Correct 1 ms 1868 KB Output is correct
6 Correct 1 ms 1868 KB Output is correct
7 Correct 1 ms 1868 KB Output is correct
8 Correct 1 ms 1872 KB Output is correct
9 Correct 1 ms 1868 KB Output is correct
10 Correct 1 ms 1868 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 1868 KB Output is correct
2 Correct 1 ms 1868 KB Output is correct
3 Correct 7 ms 1972 KB Output is correct
4 Correct 51 ms 2424 KB Output is correct
5 Correct 99 ms 2928 KB Output is correct
6 Correct 156 ms 3316 KB Output is correct
7 Correct 163 ms 3396 KB Output is correct
8 Correct 158 ms 3276 KB Output is correct
9 Correct 160 ms 3320 KB Output is correct
10 Correct 150 ms 3320 KB Output is correct
11 Correct 161 ms 3324 KB Output is correct
12 Correct 159 ms 3320 KB Output is correct
13 Correct 164 ms 3316 KB Output is correct
14 Correct 151 ms 3320 KB Output is correct
15 Correct 155 ms 3340 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 1868 KB Output is correct
2 Correct 1 ms 1868 KB Output is correct
3 Correct 7 ms 1972 KB Output is correct
4 Correct 51 ms 2424 KB Output is correct
5 Correct 99 ms 2928 KB Output is correct
6 Correct 156 ms 3316 KB Output is correct
7 Correct 163 ms 3396 KB Output is correct
8 Correct 158 ms 3276 KB Output is correct
9 Correct 160 ms 3320 KB Output is correct
10 Correct 150 ms 3320 KB Output is correct
11 Correct 161 ms 3324 KB Output is correct
12 Correct 159 ms 3320 KB Output is correct
13 Correct 164 ms 3316 KB Output is correct
14 Correct 151 ms 3320 KB Output is correct
15 Correct 155 ms 3340 KB Output is correct
16 Correct 1 ms 1868 KB Output is correct
17 Correct 2 ms 1868 KB Output is correct
18 Correct 7 ms 1868 KB Output is correct
19 Correct 59 ms 2508 KB Output is correct
20 Correct 116 ms 2928 KB Output is correct
21 Correct 177 ms 3396 KB Output is correct
22 Correct 168 ms 3320 KB Output is correct
23 Correct 171 ms 3324 KB Output is correct
24 Correct 190 ms 3320 KB Output is correct
25 Correct 171 ms 3320 KB Output is correct
26 Correct 180 ms 3220 KB Output is correct
27 Correct 202 ms 3300 KB Output is correct
28 Correct 161 ms 3396 KB Output is correct
29 Correct 173 ms 3324 KB Output is correct
30 Correct 185 ms 3296 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 1868 KB Output is correct
2 Correct 1 ms 1868 KB Output is correct
3 Correct 7 ms 1972 KB Output is correct
4 Correct 51 ms 2424 KB Output is correct
5 Correct 99 ms 2928 KB Output is correct
6 Correct 156 ms 3316 KB Output is correct
7 Correct 163 ms 3396 KB Output is correct
8 Correct 158 ms 3276 KB Output is correct
9 Correct 160 ms 3320 KB Output is correct
10 Correct 150 ms 3320 KB Output is correct
11 Correct 161 ms 3324 KB Output is correct
12 Correct 159 ms 3320 KB Output is correct
13 Correct 164 ms 3316 KB Output is correct
14 Correct 151 ms 3320 KB Output is correct
15 Correct 155 ms 3340 KB Output is correct
16 Correct 1 ms 1868 KB Output is correct
17 Correct 2 ms 1868 KB Output is correct
18 Correct 7 ms 1868 KB Output is correct
19 Correct 59 ms 2508 KB Output is correct
20 Correct 116 ms 2928 KB Output is correct
21 Correct 177 ms 3396 KB Output is correct
22 Correct 168 ms 3320 KB Output is correct
23 Correct 171 ms 3324 KB Output is correct
24 Correct 190 ms 3320 KB Output is correct
25 Correct 171 ms 3320 KB Output is correct
26 Correct 180 ms 3220 KB Output is correct
27 Correct 202 ms 3300 KB Output is correct
28 Correct 161 ms 3396 KB Output is correct
29 Correct 173 ms 3324 KB Output is correct
30 Correct 185 ms 3296 KB Output is correct
31 Correct 3 ms 1868 KB Output is correct
32 Correct 2 ms 1868 KB Output is correct
33 Correct 2 ms 1868 KB Output is correct
34 Correct 7 ms 1996 KB Output is correct
35 Correct 8 ms 2024 KB Output is correct
36 Correct 8 ms 2124 KB Output is correct
37 Correct 35 ms 2252 KB Output is correct
38 Correct 37 ms 2444 KB Output is correct
39 Correct 69 ms 2656 KB Output is correct
40 Correct 149 ms 3020 KB Output is correct
41 Correct 222 ms 3500 KB Output is correct
42 Correct 211 ms 3488 KB Output is correct
43 Correct 205 ms 3484 KB Output is correct
44 Correct 212 ms 3524 KB Output is correct
45 Correct 200 ms 3488 KB Output is correct
46 Correct 216 ms 3484 KB Output is correct
47 Correct 209 ms 3488 KB Output is correct
48 Correct 221 ms 3476 KB Output is correct
49 Correct 203 ms 3484 KB Output is correct
50 Correct 209 ms 3484 KB Output is correct
51 Correct 205 ms 3460 KB Output is correct
52 Correct 226 ms 3484 KB Output is correct
53 Correct 208 ms 3488 KB Output is correct
54 Correct 205 ms 3488 KB Output is correct
55 Correct 231 ms 3632 KB Output is correct
56 Correct 207 ms 3428 KB Output is correct
57 Correct 204 ms 3488 KB Output is correct
58 Correct 226 ms 3572 KB Output is correct
59 Correct 201 ms 3484 KB Output is correct
60 Correct 214 ms 3396 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 1740 KB Output is correct
2 Correct 1 ms 1868 KB Output is correct
3 Correct 2 ms 1868 KB Output is correct
4 Correct 1 ms 1868 KB Output is correct
5 Correct 1 ms 1868 KB Output is correct
6 Correct 1 ms 1868 KB Output is correct
7 Correct 1 ms 1868 KB Output is correct
8 Correct 1 ms 1872 KB Output is correct
9 Correct 1 ms 1868 KB Output is correct
10 Correct 1 ms 1868 KB Output is correct
11 Correct 1 ms 1868 KB Output is correct
12 Correct 1 ms 1868 KB Output is correct
13 Correct 7 ms 1972 KB Output is correct
14 Correct 51 ms 2424 KB Output is correct
15 Correct 99 ms 2928 KB Output is correct
16 Correct 156 ms 3316 KB Output is correct
17 Correct 163 ms 3396 KB Output is correct
18 Correct 158 ms 3276 KB Output is correct
19 Correct 160 ms 3320 KB Output is correct
20 Correct 150 ms 3320 KB Output is correct
21 Correct 161 ms 3324 KB Output is correct
22 Correct 159 ms 3320 KB Output is correct
23 Correct 164 ms 3316 KB Output is correct
24 Correct 151 ms 3320 KB Output is correct
25 Correct 155 ms 3340 KB Output is correct
26 Correct 1 ms 1868 KB Output is correct
27 Correct 2 ms 1868 KB Output is correct
28 Correct 7 ms 1868 KB Output is correct
29 Correct 59 ms 2508 KB Output is correct
30 Correct 116 ms 2928 KB Output is correct
31 Correct 177 ms 3396 KB Output is correct
32 Correct 168 ms 3320 KB Output is correct
33 Correct 171 ms 3324 KB Output is correct
34 Correct 190 ms 3320 KB Output is correct
35 Correct 171 ms 3320 KB Output is correct
36 Correct 180 ms 3220 KB Output is correct
37 Correct 202 ms 3300 KB Output is correct
38 Correct 161 ms 3396 KB Output is correct
39 Correct 173 ms 3324 KB Output is correct
40 Correct 185 ms 3296 KB Output is correct
41 Correct 3 ms 1868 KB Output is correct
42 Correct 2 ms 1868 KB Output is correct
43 Correct 2 ms 1868 KB Output is correct
44 Correct 7 ms 1996 KB Output is correct
45 Correct 8 ms 2024 KB Output is correct
46 Correct 8 ms 2124 KB Output is correct
47 Correct 35 ms 2252 KB Output is correct
48 Correct 37 ms 2444 KB Output is correct
49 Correct 69 ms 2656 KB Output is correct
50 Correct 149 ms 3020 KB Output is correct
51 Correct 222 ms 3500 KB Output is correct
52 Correct 211 ms 3488 KB Output is correct
53 Correct 205 ms 3484 KB Output is correct
54 Correct 212 ms 3524 KB Output is correct
55 Correct 200 ms 3488 KB Output is correct
56 Correct 216 ms 3484 KB Output is correct
57 Correct 209 ms 3488 KB Output is correct
58 Correct 221 ms 3476 KB Output is correct
59 Correct 203 ms 3484 KB Output is correct
60 Correct 209 ms 3484 KB Output is correct
61 Correct 205 ms 3460 KB Output is correct
62 Correct 226 ms 3484 KB Output is correct
63 Correct 208 ms 3488 KB Output is correct
64 Correct 205 ms 3488 KB Output is correct
65 Correct 231 ms 3632 KB Output is correct
66 Correct 207 ms 3428 KB Output is correct
67 Correct 204 ms 3488 KB Output is correct
68 Correct 226 ms 3572 KB Output is correct
69 Correct 201 ms 3484 KB Output is correct
70 Correct 214 ms 3396 KB Output is correct
71 Correct 36 ms 2664 KB Output is correct
72 Correct 50 ms 2636 KB Output is correct
73 Correct 36 ms 2644 KB Output is correct
74 Correct 36 ms 2636 KB Output is correct
75 Correct 36 ms 2636 KB Output is correct
76 Correct 73 ms 3396 KB Output is correct
77 Correct 76 ms 3500 KB Output is correct
78 Correct 73 ms 3404 KB Output is correct
79 Correct 78 ms 3344 KB Output is correct
80 Correct 90 ms 3312 KB Output is correct
81 Correct 148 ms 4568 KB Output is correct
82 Correct 152 ms 4652 KB Output is correct
83 Correct 168 ms 4604 KB Output is correct
84 Correct 145 ms 4632 KB Output is correct
85 Correct 146 ms 4624 KB Output is correct
86 Correct 161 ms 6320 KB Output is correct
87 Correct 158 ms 6356 KB Output is correct
88 Correct 157 ms 6268 KB Output is correct
89 Correct 161 ms 6348 KB Output is correct
90 Correct 156 ms 6300 KB Output is correct
91 Correct 232 ms 20472 KB Output is correct
92 Correct 243 ms 20384 KB Output is correct
93 Correct 230 ms 20408 KB Output is correct
94 Correct 236 ms 20464 KB Output is correct
95 Correct 232 ms 20420 KB Output is correct
96 Correct 230 ms 37624 KB Output is correct
97 Correct 250 ms 37612 KB Output is correct
98 Correct 241 ms 37652 KB Output is correct
99 Correct 253 ms 37688 KB Output is correct
100 Correct 236 ms 37620 KB Output is correct
101 Correct 237 ms 37528 KB Output is correct
102 Correct 237 ms 37580 KB Output is correct
103 Correct 237 ms 37564 KB Output is correct
104 Correct 250 ms 37672 KB Output is correct
105 Correct 240 ms 37596 KB Output is correct
106 Correct 238 ms 46136 KB Output is correct
107 Correct 260 ms 46192 KB Output is correct
108 Correct 236 ms 46188 KB Output is correct
109 Correct 272 ms 46180 KB Output is correct
110 Correct 251 ms 46140 KB Output is correct
111 Correct 242 ms 46176 KB Output is correct
112 Correct 239 ms 46080 KB Output is correct
113 Correct 253 ms 46332 KB Output is correct
114 Correct 237 ms 46088 KB Output is correct
115 Correct 239 ms 46164 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 1740 KB Output is correct
2 Correct 1 ms 1868 KB Output is correct
3 Correct 2 ms 1868 KB Output is correct
4 Correct 1 ms 1868 KB Output is correct
5 Correct 1 ms 1868 KB Output is correct
6 Correct 1 ms 1868 KB Output is correct
7 Correct 1 ms 1868 KB Output is correct
8 Correct 1 ms 1872 KB Output is correct
9 Correct 1 ms 1868 KB Output is correct
10 Correct 1 ms 1868 KB Output is correct
11 Correct 1 ms 1868 KB Output is correct
12 Correct 1 ms 1868 KB Output is correct
13 Correct 7 ms 1972 KB Output is correct
14 Correct 51 ms 2424 KB Output is correct
15 Correct 99 ms 2928 KB Output is correct
16 Correct 156 ms 3316 KB Output is correct
17 Correct 163 ms 3396 KB Output is correct
18 Correct 158 ms 3276 KB Output is correct
19 Correct 160 ms 3320 KB Output is correct
20 Correct 150 ms 3320 KB Output is correct
21 Correct 161 ms 3324 KB Output is correct
22 Correct 159 ms 3320 KB Output is correct
23 Correct 164 ms 3316 KB Output is correct
24 Correct 151 ms 3320 KB Output is correct
25 Correct 155 ms 3340 KB Output is correct
26 Correct 1 ms 1868 KB Output is correct
27 Correct 2 ms 1868 KB Output is correct
28 Correct 7 ms 1868 KB Output is correct
29 Correct 59 ms 2508 KB Output is correct
30 Correct 116 ms 2928 KB Output is correct
31 Correct 177 ms 3396 KB Output is correct
32 Correct 168 ms 3320 KB Output is correct
33 Correct 171 ms 3324 KB Output is correct
34 Correct 190 ms 3320 KB Output is correct
35 Correct 171 ms 3320 KB Output is correct
36 Correct 180 ms 3220 KB Output is correct
37 Correct 202 ms 3300 KB Output is correct
38 Correct 161 ms 3396 KB Output is correct
39 Correct 173 ms 3324 KB Output is correct
40 Correct 185 ms 3296 KB Output is correct
41 Correct 3 ms 1868 KB Output is correct
42 Correct 2 ms 1868 KB Output is correct
43 Correct 2 ms 1868 KB Output is correct
44 Correct 7 ms 1996 KB Output is correct
45 Correct 8 ms 2024 KB Output is correct
46 Correct 8 ms 2124 KB Output is correct
47 Correct 35 ms 2252 KB Output is correct
48 Correct 37 ms 2444 KB Output is correct
49 Correct 69 ms 2656 KB Output is correct
50 Correct 149 ms 3020 KB Output is correct
51 Correct 222 ms 3500 KB Output is correct
52 Correct 211 ms 3488 KB Output is correct
53 Correct 205 ms 3484 KB Output is correct
54 Correct 212 ms 3524 KB Output is correct
55 Correct 200 ms 3488 KB Output is correct
56 Correct 216 ms 3484 KB Output is correct
57 Correct 209 ms 3488 KB Output is correct
58 Correct 221 ms 3476 KB Output is correct
59 Correct 203 ms 3484 KB Output is correct
60 Correct 209 ms 3484 KB Output is correct
61 Correct 205 ms 3460 KB Output is correct
62 Correct 226 ms 3484 KB Output is correct
63 Correct 208 ms 3488 KB Output is correct
64 Correct 205 ms 3488 KB Output is correct
65 Correct 231 ms 3632 KB Output is correct
66 Correct 207 ms 3428 KB Output is correct
67 Correct 204 ms 3488 KB Output is correct
68 Correct 226 ms 3572 KB Output is correct
69 Correct 201 ms 3484 KB Output is correct
70 Correct 214 ms 3396 KB Output is correct
71 Correct 36 ms 2664 KB Output is correct
72 Correct 50 ms 2636 KB Output is correct
73 Correct 36 ms 2644 KB Output is correct
74 Correct 36 ms 2636 KB Output is correct
75 Correct 36 ms 2636 KB Output is correct
76 Correct 73 ms 3396 KB Output is correct
77 Correct 76 ms 3500 KB Output is correct
78 Correct 73 ms 3404 KB Output is correct
79 Correct 78 ms 3344 KB Output is correct
80 Correct 90 ms 3312 KB Output is correct
81 Correct 148 ms 4568 KB Output is correct
82 Correct 152 ms 4652 KB Output is correct
83 Correct 168 ms 4604 KB Output is correct
84 Correct 145 ms 4632 KB Output is correct
85 Correct 146 ms 4624 KB Output is correct
86 Correct 161 ms 6320 KB Output is correct
87 Correct 158 ms 6356 KB Output is correct
88 Correct 157 ms 6268 KB Output is correct
89 Correct 161 ms 6348 KB Output is correct
90 Correct 156 ms 6300 KB Output is correct
91 Correct 232 ms 20472 KB Output is correct
92 Correct 243 ms 20384 KB Output is correct
93 Correct 230 ms 20408 KB Output is correct
94 Correct 236 ms 20464 KB Output is correct
95 Correct 232 ms 20420 KB Output is correct
96 Correct 230 ms 37624 KB Output is correct
97 Correct 250 ms 37612 KB Output is correct
98 Correct 241 ms 37652 KB Output is correct
99 Correct 253 ms 37688 KB Output is correct
100 Correct 236 ms 37620 KB Output is correct
101 Correct 237 ms 37528 KB Output is correct
102 Correct 237 ms 37580 KB Output is correct
103 Correct 237 ms 37564 KB Output is correct
104 Correct 250 ms 37672 KB Output is correct
105 Correct 240 ms 37596 KB Output is correct
106 Correct 238 ms 46136 KB Output is correct
107 Correct 260 ms 46192 KB Output is correct
108 Correct 236 ms 46188 KB Output is correct
109 Correct 272 ms 46180 KB Output is correct
110 Correct 251 ms 46140 KB Output is correct
111 Correct 242 ms 46176 KB Output is correct
112 Correct 239 ms 46080 KB Output is correct
113 Correct 253 ms 46332 KB Output is correct
114 Correct 237 ms 46088 KB Output is correct
115 Correct 239 ms 46164 KB Output is correct
116 Correct 4809 ms 3940 KB Output is correct
117 Correct 4917 ms 3836 KB Output is correct
118 Execution timed out 7005 ms 3656 KB Time limit exceeded
119 Halted 0 ms 0 KB -