#include "shoes.h"
#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> pi;
const int N=2e5+10;
int fw[N], fw2[N];
void update(int x, int y, int v) { //inclusive
++x; ++y;
for (int tx=x; tx < N; tx += tx&(-tx)) fw[tx] += v, fw2[tx] -= v*(x-1);
for (int ty=y+1; ty < N; ty += ty&(-ty)) fw[ty] -= v, fw2[ty] += v*y;
}
int sum (int x) {
++x;
int res = 0;
for (int tx=x; tx; tx -= tx&(-tx)) res += fw[tx]*x + fw2[tx];
return res;
}
int range_sum(int x, int y) { //inclusive
return sum(y)-sum(x-1);
}
long long count_swaps(vector<int> s) {
int n=s.size();
deque<int> dq[n/2+1], dq2[n/2+1];
for(int i=0; i<n; ++i) {
if(s[i]<0) dq[-s[i]].push_back(i+1);
else dq2[s[i]].push_back(i+1);
}
for(int i=0; i<=n; ++i) {
update(i+1,i+1,i+1);
}
long long ans=0;
vector<pair<int,int>> pairs;
for(int i=1; i<=n/2; ++i) {
for(int j=0; j<dq[i].size(); ++j) {
if(dq[i][j]>dq2[i][j]) {
++ans;
swap(dq[i][j], dq2[i][j]);
}
pairs.emplace_back(dq[i][j], dq2[i][j]);
}
}
sort(pairs.begin(), pairs.end());
long long cnt=0;
for(auto it: pairs) {
int l=it.first, r=it.second;
ans += range_sum(r,r) + range_sum(l,l) - (cnt*2+1) - (cnt*2+2); //<< '\n';
++cnt;
update(0,l,1); update(0,r,1);
}
return ans;
}
// int main() {
// cout << count_swaps({2,1,-1,-2});
// }
/*
dq[1] = [2]
dq[2] = [1]
dq2[1] = [3]
dq2[2] = [4]
*/
Compilation message
shoes.cpp: In function 'long long int count_swaps(std::vector<int>)':
shoes.cpp:38:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::deque<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
38 | for(int j=0; j<dq[i].size(); ++j) {
| ~^~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
308 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
308 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
340 KB |
Output is correct |
5 |
Correct |
1 ms |
304 KB |
Output is correct |
6 |
Correct |
1 ms |
312 KB |
Output is correct |
7 |
Correct |
1 ms |
308 KB |
Output is correct |
8 |
Correct |
1 ms |
340 KB |
Output is correct |
9 |
Correct |
0 ms |
340 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Correct |
1 ms |
340 KB |
Output is correct |
12 |
Correct |
1 ms |
312 KB |
Output is correct |
13 |
Correct |
1 ms |
340 KB |
Output is correct |
14 |
Correct |
1 ms |
308 KB |
Output is correct |
15 |
Correct |
1 ms |
340 KB |
Output is correct |
16 |
Correct |
1 ms |
312 KB |
Output is correct |
17 |
Correct |
1 ms |
340 KB |
Output is correct |
18 |
Correct |
1 ms |
340 KB |
Output is correct |
19 |
Correct |
1 ms |
340 KB |
Output is correct |
20 |
Correct |
1 ms |
340 KB |
Output is correct |
21 |
Correct |
1 ms |
312 KB |
Output is correct |
22 |
Correct |
1 ms |
340 KB |
Output is correct |
23 |
Correct |
1 ms |
340 KB |
Output is correct |
24 |
Correct |
1 ms |
308 KB |
Output is correct |
25 |
Correct |
1 ms |
312 KB |
Output is correct |
26 |
Correct |
1 ms |
340 KB |
Output is correct |
27 |
Correct |
1 ms |
340 KB |
Output is correct |
28 |
Correct |
1 ms |
340 KB |
Output is correct |
29 |
Correct |
1 ms |
308 KB |
Output is correct |
30 |
Correct |
1 ms |
340 KB |
Output is correct |
31 |
Correct |
1 ms |
340 KB |
Output is correct |
32 |
Correct |
1 ms |
304 KB |
Output is correct |
33 |
Correct |
1 ms |
340 KB |
Output is correct |
34 |
Correct |
1 ms |
308 KB |
Output is correct |
35 |
Correct |
1 ms |
340 KB |
Output is correct |
36 |
Correct |
1 ms |
304 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
308 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
340 KB |
Output is correct |
5 |
Correct |
1 ms |
340 KB |
Output is correct |
6 |
Correct |
1 ms |
340 KB |
Output is correct |
7 |
Correct |
1 ms |
304 KB |
Output is correct |
8 |
Correct |
1 ms |
340 KB |
Output is correct |
9 |
Correct |
1 ms |
340 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Correct |
1 ms |
340 KB |
Output is correct |
12 |
Correct |
1 ms |
308 KB |
Output is correct |
13 |
Correct |
1 ms |
304 KB |
Output is correct |
14 |
Correct |
1 ms |
304 KB |
Output is correct |
15 |
Correct |
0 ms |
340 KB |
Output is correct |
16 |
Correct |
1 ms |
468 KB |
Output is correct |
17 |
Correct |
1 ms |
692 KB |
Output is correct |
18 |
Correct |
2 ms |
1748 KB |
Output is correct |
19 |
Correct |
2 ms |
1748 KB |
Output is correct |
20 |
Correct |
16 ms |
14396 KB |
Output is correct |
21 |
Correct |
14 ms |
14420 KB |
Output is correct |
22 |
Correct |
158 ms |
141272 KB |
Output is correct |
23 |
Correct |
139 ms |
141260 KB |
Output is correct |
24 |
Correct |
142 ms |
141232 KB |
Output is correct |
25 |
Correct |
137 ms |
141044 KB |
Output is correct |
26 |
Correct |
133 ms |
141292 KB |
Output is correct |
27 |
Correct |
137 ms |
141064 KB |
Output is correct |
28 |
Correct |
129 ms |
141228 KB |
Output is correct |
29 |
Correct |
130 ms |
141320 KB |
Output is correct |
30 |
Correct |
129 ms |
141232 KB |
Output is correct |
31 |
Correct |
139 ms |
141224 KB |
Output is correct |
32 |
Correct |
129 ms |
141236 KB |
Output is correct |
33 |
Correct |
131 ms |
141108 KB |
Output is correct |
34 |
Correct |
137 ms |
141240 KB |
Output is correct |
35 |
Correct |
1 ms |
304 KB |
Output is correct |
36 |
Correct |
1 ms |
340 KB |
Output is correct |
37 |
Correct |
127 ms |
141232 KB |
Output is correct |
38 |
Correct |
142 ms |
141272 KB |
Output is correct |
39 |
Correct |
137 ms |
141324 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
340 KB |
Output is correct |
3 |
Correct |
1 ms |
312 KB |
Output is correct |
4 |
Correct |
1 ms |
304 KB |
Output is correct |
5 |
Correct |
152 ms |
140416 KB |
Output is correct |
6 |
Correct |
146 ms |
140420 KB |
Output is correct |
7 |
Correct |
179 ms |
140472 KB |
Output is correct |
8 |
Correct |
145 ms |
141316 KB |
Output is correct |
9 |
Correct |
160 ms |
140436 KB |
Output is correct |
10 |
Correct |
171 ms |
140588 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
308 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
340 KB |
Output is correct |
5 |
Correct |
1 ms |
304 KB |
Output is correct |
6 |
Correct |
1 ms |
312 KB |
Output is correct |
7 |
Correct |
1 ms |
308 KB |
Output is correct |
8 |
Correct |
1 ms |
340 KB |
Output is correct |
9 |
Correct |
0 ms |
340 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Correct |
1 ms |
340 KB |
Output is correct |
12 |
Correct |
1 ms |
312 KB |
Output is correct |
13 |
Correct |
1 ms |
340 KB |
Output is correct |
14 |
Correct |
1 ms |
308 KB |
Output is correct |
15 |
Correct |
1 ms |
340 KB |
Output is correct |
16 |
Correct |
1 ms |
312 KB |
Output is correct |
17 |
Correct |
1 ms |
340 KB |
Output is correct |
18 |
Correct |
1 ms |
340 KB |
Output is correct |
19 |
Correct |
1 ms |
340 KB |
Output is correct |
20 |
Correct |
1 ms |
340 KB |
Output is correct |
21 |
Correct |
1 ms |
312 KB |
Output is correct |
22 |
Correct |
1 ms |
340 KB |
Output is correct |
23 |
Correct |
1 ms |
340 KB |
Output is correct |
24 |
Correct |
1 ms |
308 KB |
Output is correct |
25 |
Correct |
1 ms |
312 KB |
Output is correct |
26 |
Correct |
1 ms |
340 KB |
Output is correct |
27 |
Correct |
1 ms |
340 KB |
Output is correct |
28 |
Correct |
1 ms |
340 KB |
Output is correct |
29 |
Correct |
1 ms |
308 KB |
Output is correct |
30 |
Correct |
1 ms |
340 KB |
Output is correct |
31 |
Correct |
1 ms |
340 KB |
Output is correct |
32 |
Correct |
1 ms |
304 KB |
Output is correct |
33 |
Correct |
1 ms |
340 KB |
Output is correct |
34 |
Correct |
1 ms |
308 KB |
Output is correct |
35 |
Correct |
1 ms |
340 KB |
Output is correct |
36 |
Correct |
1 ms |
304 KB |
Output is correct |
37 |
Correct |
1 ms |
340 KB |
Output is correct |
38 |
Correct |
1 ms |
340 KB |
Output is correct |
39 |
Correct |
1 ms |
468 KB |
Output is correct |
40 |
Correct |
1 ms |
724 KB |
Output is correct |
41 |
Correct |
2 ms |
1748 KB |
Output is correct |
42 |
Correct |
2 ms |
1748 KB |
Output is correct |
43 |
Correct |
2 ms |
1720 KB |
Output is correct |
44 |
Correct |
2 ms |
1748 KB |
Output is correct |
45 |
Correct |
2 ms |
1748 KB |
Output is correct |
46 |
Correct |
2 ms |
1748 KB |
Output is correct |
47 |
Correct |
2 ms |
1748 KB |
Output is correct |
48 |
Correct |
2 ms |
1748 KB |
Output is correct |
49 |
Correct |
3 ms |
1748 KB |
Output is correct |
50 |
Correct |
2 ms |
1720 KB |
Output is correct |
51 |
Correct |
3 ms |
1748 KB |
Output is correct |
52 |
Correct |
4 ms |
1748 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
340 KB |
Output is correct |
2 |
Correct |
1 ms |
308 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
340 KB |
Output is correct |
5 |
Correct |
1 ms |
304 KB |
Output is correct |
6 |
Correct |
1 ms |
312 KB |
Output is correct |
7 |
Correct |
1 ms |
308 KB |
Output is correct |
8 |
Correct |
1 ms |
340 KB |
Output is correct |
9 |
Correct |
0 ms |
340 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Correct |
1 ms |
340 KB |
Output is correct |
12 |
Correct |
1 ms |
312 KB |
Output is correct |
13 |
Correct |
1 ms |
340 KB |
Output is correct |
14 |
Correct |
1 ms |
308 KB |
Output is correct |
15 |
Correct |
1 ms |
340 KB |
Output is correct |
16 |
Correct |
1 ms |
312 KB |
Output is correct |
17 |
Correct |
1 ms |
340 KB |
Output is correct |
18 |
Correct |
1 ms |
340 KB |
Output is correct |
19 |
Correct |
1 ms |
340 KB |
Output is correct |
20 |
Correct |
1 ms |
340 KB |
Output is correct |
21 |
Correct |
1 ms |
312 KB |
Output is correct |
22 |
Correct |
1 ms |
340 KB |
Output is correct |
23 |
Correct |
1 ms |
340 KB |
Output is correct |
24 |
Correct |
1 ms |
308 KB |
Output is correct |
25 |
Correct |
1 ms |
312 KB |
Output is correct |
26 |
Correct |
1 ms |
340 KB |
Output is correct |
27 |
Correct |
1 ms |
340 KB |
Output is correct |
28 |
Correct |
1 ms |
340 KB |
Output is correct |
29 |
Correct |
1 ms |
308 KB |
Output is correct |
30 |
Correct |
1 ms |
340 KB |
Output is correct |
31 |
Correct |
1 ms |
340 KB |
Output is correct |
32 |
Correct |
1 ms |
304 KB |
Output is correct |
33 |
Correct |
1 ms |
340 KB |
Output is correct |
34 |
Correct |
1 ms |
308 KB |
Output is correct |
35 |
Correct |
1 ms |
340 KB |
Output is correct |
36 |
Correct |
1 ms |
304 KB |
Output is correct |
37 |
Correct |
1 ms |
340 KB |
Output is correct |
38 |
Correct |
1 ms |
340 KB |
Output is correct |
39 |
Correct |
1 ms |
340 KB |
Output is correct |
40 |
Correct |
1 ms |
340 KB |
Output is correct |
41 |
Correct |
1 ms |
304 KB |
Output is correct |
42 |
Correct |
1 ms |
340 KB |
Output is correct |
43 |
Correct |
1 ms |
340 KB |
Output is correct |
44 |
Correct |
1 ms |
340 KB |
Output is correct |
45 |
Correct |
1 ms |
340 KB |
Output is correct |
46 |
Correct |
1 ms |
308 KB |
Output is correct |
47 |
Correct |
1 ms |
304 KB |
Output is correct |
48 |
Correct |
1 ms |
304 KB |
Output is correct |
49 |
Correct |
0 ms |
340 KB |
Output is correct |
50 |
Correct |
1 ms |
468 KB |
Output is correct |
51 |
Correct |
1 ms |
692 KB |
Output is correct |
52 |
Correct |
2 ms |
1748 KB |
Output is correct |
53 |
Correct |
2 ms |
1748 KB |
Output is correct |
54 |
Correct |
16 ms |
14396 KB |
Output is correct |
55 |
Correct |
14 ms |
14420 KB |
Output is correct |
56 |
Correct |
158 ms |
141272 KB |
Output is correct |
57 |
Correct |
139 ms |
141260 KB |
Output is correct |
58 |
Correct |
142 ms |
141232 KB |
Output is correct |
59 |
Correct |
137 ms |
141044 KB |
Output is correct |
60 |
Correct |
133 ms |
141292 KB |
Output is correct |
61 |
Correct |
137 ms |
141064 KB |
Output is correct |
62 |
Correct |
129 ms |
141228 KB |
Output is correct |
63 |
Correct |
130 ms |
141320 KB |
Output is correct |
64 |
Correct |
129 ms |
141232 KB |
Output is correct |
65 |
Correct |
139 ms |
141224 KB |
Output is correct |
66 |
Correct |
129 ms |
141236 KB |
Output is correct |
67 |
Correct |
131 ms |
141108 KB |
Output is correct |
68 |
Correct |
137 ms |
141240 KB |
Output is correct |
69 |
Correct |
1 ms |
304 KB |
Output is correct |
70 |
Correct |
1 ms |
340 KB |
Output is correct |
71 |
Correct |
127 ms |
141232 KB |
Output is correct |
72 |
Correct |
142 ms |
141272 KB |
Output is correct |
73 |
Correct |
137 ms |
141324 KB |
Output is correct |
74 |
Correct |
1 ms |
340 KB |
Output is correct |
75 |
Correct |
1 ms |
340 KB |
Output is correct |
76 |
Correct |
1 ms |
312 KB |
Output is correct |
77 |
Correct |
1 ms |
304 KB |
Output is correct |
78 |
Correct |
152 ms |
140416 KB |
Output is correct |
79 |
Correct |
146 ms |
140420 KB |
Output is correct |
80 |
Correct |
179 ms |
140472 KB |
Output is correct |
81 |
Correct |
145 ms |
141316 KB |
Output is correct |
82 |
Correct |
160 ms |
140436 KB |
Output is correct |
83 |
Correct |
171 ms |
140588 KB |
Output is correct |
84 |
Correct |
1 ms |
340 KB |
Output is correct |
85 |
Correct |
1 ms |
340 KB |
Output is correct |
86 |
Correct |
1 ms |
468 KB |
Output is correct |
87 |
Correct |
1 ms |
724 KB |
Output is correct |
88 |
Correct |
2 ms |
1748 KB |
Output is correct |
89 |
Correct |
2 ms |
1748 KB |
Output is correct |
90 |
Correct |
2 ms |
1720 KB |
Output is correct |
91 |
Correct |
2 ms |
1748 KB |
Output is correct |
92 |
Correct |
2 ms |
1748 KB |
Output is correct |
93 |
Correct |
2 ms |
1748 KB |
Output is correct |
94 |
Correct |
2 ms |
1748 KB |
Output is correct |
95 |
Correct |
2 ms |
1748 KB |
Output is correct |
96 |
Correct |
3 ms |
1748 KB |
Output is correct |
97 |
Correct |
2 ms |
1720 KB |
Output is correct |
98 |
Correct |
3 ms |
1748 KB |
Output is correct |
99 |
Correct |
4 ms |
1748 KB |
Output is correct |
100 |
Correct |
2 ms |
1748 KB |
Output is correct |
101 |
Correct |
16 ms |
14404 KB |
Output is correct |
102 |
Correct |
16 ms |
14420 KB |
Output is correct |
103 |
Correct |
194 ms |
140432 KB |
Output is correct |
104 |
Correct |
164 ms |
140436 KB |
Output is correct |
105 |
Correct |
182 ms |
140412 KB |
Output is correct |
106 |
Correct |
164 ms |
140424 KB |
Output is correct |
107 |
Correct |
155 ms |
140396 KB |
Output is correct |
108 |
Correct |
151 ms |
140476 KB |
Output is correct |
109 |
Correct |
142 ms |
141256 KB |
Output is correct |
110 |
Correct |
133 ms |
141228 KB |
Output is correct |
111 |
Correct |
161 ms |
140456 KB |
Output is correct |
112 |
Correct |
173 ms |
140432 KB |
Output is correct |
113 |
Correct |
155 ms |
140372 KB |
Output is correct |
114 |
Correct |
156 ms |
140368 KB |
Output is correct |
115 |
Correct |
159 ms |
140376 KB |
Output is correct |