#include "swap.h"
#include <bits/stdc++.h>
#include <vector>
using namespace std;
long long n, m, i, j, sz[200001], p[200001], d[200001], mx[200001], szm[200001], kek[200001];
multiset<pair<long long, long long>> ms;
int mxx;
bool st1, st2;
vector<pair<long long, long long>> g[200001];
vector<pair<long long, pair<long long, long long>>> gg;
long long get(long long x) {
return x == p[x] ? x : p[x] = get(p[x]);
}
void merge(long long xx, long long yy) {
long long x = get(xx), y = get(yy);
mx[x] = max({mx[x], d[x], d[xx]});
mx[y] = max({mx[y], d[y], d[yy]});
mx[x] = max(mx[x], mx[y]);
mx[y] = max(mx[y], mx[x]);
szm[x]++;
if (x == y) return;
if (sz[x] < sz[y]) swap(x, y);
p[y] = x;
sz[x] += sz[y];
szm[x] += szm[y];
mx[x] = max(mx[x], mx[y]);
mx[y] = max(mx[y], mx[x]);
}
void init(int N, int M,
std::vector<int> U, std::vector<int> V, std::vector<int> W) {
n = N, m = M;
st2 = 1;
for (long long i = 0; i < M; i++) {
g[U[i]].push_back({V[i], W[i]});
g[V[i]].push_back({U[i], W[i]});
if (U[i]) {
st2 = 0;
}
gg.push_back({W[i], {U[i], V[i]}});
kek[V[i]] = W[i];
mxx = max(mxx, W[i]);
}
if (st2) {
for (long long i = 0; i < N; i++) ms.insert({kek[i], i});
}
st1 = 1;
for (long long i = 0; i < N; i++) {
if (g[i].size() > 2) st1 = 0;
}
sort(gg.begin(), gg.end());
}
bool check(long long md, long long x, long long y) {
if (md < 0 || md >= gg.size()) return 0;
for (long long i = 0; i < n; i++) sz[i] = 1, p[i] = i, d[i] = mx[i] = szm[i] = 0;
for (long long i = 0; i <= md; i++) {
d[gg[i].second.first]++;
d[gg[i].second.second]++;
merge(gg[i].second.first, gg[i].second.second);
}
if (get(x) != get(y)) return 0;
if (mx[get(x)] <= 2 && szm[get(x)] < sz[get(x)]) return 0;
return 1;
}
int getMinimumFuelCapacity(int X, int Y) {
if (st1) {
if (n == m) return mxx;
return -1;
}
if (st2) {
if (ms.size() > 2) {
ms.erase({kek[X], X});
ms.erase({kek[Y], Y});
long long ans = max({kek[X], kek[Y], (*ms.begin()).first});
ms.insert({kek[X], X});
ms.insert({kek[Y], Y});
return ans;
}
return -1;
}
long long l = 0, r = (int)gg.size(), md, ans = 0;
while (l <= r) {
md = (l + r) / 2;
if (check(md, X, Y)) ans = md, r = md - 1;
else l = md + 1;
}
if (!check(ans, X, Y)) return -1;
return gg[ans].first;
}
Compilation message
swap.cpp: In function 'bool check(long long int, long long int, long long int)':
swap.cpp:57:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, std::pair<long long int, long long int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
57 | if (md < 0 || md >= gg.size()) return 0;
| ~~~^~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
6744 KB |
Output is correct |
2 |
Correct |
2 ms |
6748 KB |
Output is correct |
3 |
Correct |
1 ms |
6748 KB |
Output is correct |
4 |
Correct |
2 ms |
6748 KB |
Output is correct |
5 |
Correct |
2 ms |
6748 KB |
Output is correct |
6 |
Correct |
2 ms |
6748 KB |
Output is correct |
7 |
Correct |
2 ms |
6772 KB |
Output is correct |
8 |
Correct |
2 ms |
6836 KB |
Output is correct |
9 |
Correct |
41 ms |
15080 KB |
Output is correct |
10 |
Correct |
49 ms |
16904 KB |
Output is correct |
11 |
Correct |
49 ms |
16000 KB |
Output is correct |
12 |
Correct |
65 ms |
16644 KB |
Output is correct |
13 |
Correct |
50 ms |
16644 KB |
Output is correct |
14 |
Correct |
44 ms |
15348 KB |
Output is correct |
15 |
Correct |
96 ms |
18728 KB |
Output is correct |
16 |
Correct |
102 ms |
17696 KB |
Output is correct |
17 |
Correct |
91 ms |
18540 KB |
Output is correct |
18 |
Correct |
120 ms |
18480 KB |
Output is correct |
19 |
Correct |
48 ms |
11604 KB |
Output is correct |
20 |
Correct |
89 ms |
19348 KB |
Output is correct |
21 |
Correct |
92 ms |
20008 KB |
Output is correct |
22 |
Correct |
99 ms |
19760 KB |
Output is correct |
23 |
Correct |
100 ms |
19724 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
6744 KB |
Output is correct |
2 |
Correct |
2 ms |
6748 KB |
Output is correct |
3 |
Correct |
272 ms |
24500 KB |
Output is correct |
4 |
Correct |
265 ms |
29080 KB |
Output is correct |
5 |
Incorrect |
290 ms |
29088 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
6744 KB |
Output is correct |
2 |
Correct |
2 ms |
6748 KB |
Output is correct |
3 |
Correct |
1 ms |
6748 KB |
Output is correct |
4 |
Correct |
2 ms |
6748 KB |
Output is correct |
5 |
Correct |
2 ms |
6748 KB |
Output is correct |
6 |
Correct |
2 ms |
6748 KB |
Output is correct |
7 |
Correct |
2 ms |
6772 KB |
Output is correct |
8 |
Correct |
2 ms |
6836 KB |
Output is correct |
9 |
Correct |
2 ms |
12892 KB |
Output is correct |
10 |
Correct |
3 ms |
12892 KB |
Output is correct |
11 |
Correct |
3 ms |
12892 KB |
Output is correct |
12 |
Correct |
4 ms |
12892 KB |
Output is correct |
13 |
Correct |
4 ms |
12892 KB |
Output is correct |
14 |
Correct |
4 ms |
12892 KB |
Output is correct |
15 |
Correct |
4 ms |
12888 KB |
Output is correct |
16 |
Correct |
4 ms |
12892 KB |
Output is correct |
17 |
Correct |
4 ms |
12892 KB |
Output is correct |
18 |
Correct |
3 ms |
12888 KB |
Output is correct |
19 |
Correct |
3 ms |
12892 KB |
Output is correct |
20 |
Correct |
3 ms |
12892 KB |
Output is correct |
21 |
Correct |
3 ms |
13076 KB |
Output is correct |
22 |
Correct |
3 ms |
12892 KB |
Output is correct |
23 |
Correct |
3 ms |
12892 KB |
Output is correct |
24 |
Correct |
4 ms |
12892 KB |
Output is correct |
25 |
Correct |
4 ms |
12892 KB |
Output is correct |
26 |
Correct |
4 ms |
12892 KB |
Output is correct |
27 |
Correct |
3 ms |
12888 KB |
Output is correct |
28 |
Correct |
4 ms |
12888 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
12892 KB |
Output is correct |
2 |
Correct |
1 ms |
6744 KB |
Output is correct |
3 |
Correct |
2 ms |
6748 KB |
Output is correct |
4 |
Correct |
1 ms |
6748 KB |
Output is correct |
5 |
Correct |
2 ms |
6748 KB |
Output is correct |
6 |
Correct |
2 ms |
6748 KB |
Output is correct |
7 |
Correct |
2 ms |
6748 KB |
Output is correct |
8 |
Correct |
2 ms |
6772 KB |
Output is correct |
9 |
Correct |
2 ms |
6836 KB |
Output is correct |
10 |
Correct |
41 ms |
15080 KB |
Output is correct |
11 |
Correct |
49 ms |
16904 KB |
Output is correct |
12 |
Correct |
49 ms |
16000 KB |
Output is correct |
13 |
Correct |
65 ms |
16644 KB |
Output is correct |
14 |
Correct |
50 ms |
16644 KB |
Output is correct |
15 |
Correct |
3 ms |
12892 KB |
Output is correct |
16 |
Correct |
3 ms |
12892 KB |
Output is correct |
17 |
Correct |
4 ms |
12892 KB |
Output is correct |
18 |
Correct |
4 ms |
12892 KB |
Output is correct |
19 |
Correct |
4 ms |
12892 KB |
Output is correct |
20 |
Correct |
4 ms |
12888 KB |
Output is correct |
21 |
Correct |
4 ms |
12892 KB |
Output is correct |
22 |
Correct |
4 ms |
12892 KB |
Output is correct |
23 |
Correct |
3 ms |
12888 KB |
Output is correct |
24 |
Correct |
3 ms |
12892 KB |
Output is correct |
25 |
Correct |
3 ms |
12892 KB |
Output is correct |
26 |
Correct |
3 ms |
13076 KB |
Output is correct |
27 |
Correct |
3 ms |
12892 KB |
Output is correct |
28 |
Correct |
3 ms |
12892 KB |
Output is correct |
29 |
Correct |
4 ms |
12892 KB |
Output is correct |
30 |
Correct |
4 ms |
12892 KB |
Output is correct |
31 |
Correct |
4 ms |
12892 KB |
Output is correct |
32 |
Correct |
3 ms |
12888 KB |
Output is correct |
33 |
Correct |
4 ms |
12888 KB |
Output is correct |
34 |
Correct |
12 ms |
14232 KB |
Output is correct |
35 |
Correct |
355 ms |
22484 KB |
Output is correct |
36 |
Correct |
378 ms |
22632 KB |
Output is correct |
37 |
Correct |
335 ms |
22452 KB |
Output is correct |
38 |
Correct |
338 ms |
22332 KB |
Output is correct |
39 |
Correct |
343 ms |
22300 KB |
Output is correct |
40 |
Correct |
310 ms |
21632 KB |
Output is correct |
41 |
Correct |
371 ms |
22468 KB |
Output is correct |
42 |
Correct |
369 ms |
22464 KB |
Output is correct |
43 |
Correct |
190 ms |
22468 KB |
Output is correct |
44 |
Correct |
365 ms |
22536 KB |
Output is correct |
45 |
Correct |
293 ms |
26684 KB |
Output is correct |
46 |
Correct |
364 ms |
22380 KB |
Output is correct |
47 |
Correct |
336 ms |
23176 KB |
Output is correct |
48 |
Correct |
322 ms |
23272 KB |
Output is correct |
49 |
Correct |
69 ms |
28860 KB |
Output is correct |
50 |
Correct |
66 ms |
23868 KB |
Output is correct |
51 |
Correct |
210 ms |
24624 KB |
Output is correct |
52 |
Correct |
417 ms |
29576 KB |
Output is correct |
53 |
Correct |
357 ms |
31132 KB |
Output is correct |
54 |
Correct |
418 ms |
32152 KB |
Output is correct |
55 |
Correct |
245 ms |
22372 KB |
Output is correct |
56 |
Correct |
416 ms |
29964 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
6744 KB |
Output is correct |
2 |
Correct |
2 ms |
6748 KB |
Output is correct |
3 |
Correct |
1 ms |
6748 KB |
Output is correct |
4 |
Correct |
2 ms |
6748 KB |
Output is correct |
5 |
Correct |
2 ms |
6748 KB |
Output is correct |
6 |
Correct |
2 ms |
6748 KB |
Output is correct |
7 |
Correct |
2 ms |
6772 KB |
Output is correct |
8 |
Correct |
2 ms |
6836 KB |
Output is correct |
9 |
Correct |
41 ms |
15080 KB |
Output is correct |
10 |
Correct |
49 ms |
16904 KB |
Output is correct |
11 |
Correct |
49 ms |
16000 KB |
Output is correct |
12 |
Correct |
65 ms |
16644 KB |
Output is correct |
13 |
Correct |
50 ms |
16644 KB |
Output is correct |
14 |
Correct |
44 ms |
15348 KB |
Output is correct |
15 |
Correct |
96 ms |
18728 KB |
Output is correct |
16 |
Correct |
102 ms |
17696 KB |
Output is correct |
17 |
Correct |
91 ms |
18540 KB |
Output is correct |
18 |
Correct |
120 ms |
18480 KB |
Output is correct |
19 |
Correct |
272 ms |
24500 KB |
Output is correct |
20 |
Correct |
265 ms |
29080 KB |
Output is correct |
21 |
Incorrect |
290 ms |
29088 KB |
Output isn't correct |
22 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
12892 KB |
Output is correct |
2 |
Correct |
1 ms |
6744 KB |
Output is correct |
3 |
Correct |
2 ms |
6748 KB |
Output is correct |
4 |
Correct |
1 ms |
6748 KB |
Output is correct |
5 |
Correct |
2 ms |
6748 KB |
Output is correct |
6 |
Correct |
2 ms |
6748 KB |
Output is correct |
7 |
Correct |
2 ms |
6748 KB |
Output is correct |
8 |
Correct |
2 ms |
6772 KB |
Output is correct |
9 |
Correct |
2 ms |
6836 KB |
Output is correct |
10 |
Correct |
41 ms |
15080 KB |
Output is correct |
11 |
Correct |
49 ms |
16904 KB |
Output is correct |
12 |
Correct |
49 ms |
16000 KB |
Output is correct |
13 |
Correct |
65 ms |
16644 KB |
Output is correct |
14 |
Correct |
50 ms |
16644 KB |
Output is correct |
15 |
Correct |
44 ms |
15348 KB |
Output is correct |
16 |
Correct |
96 ms |
18728 KB |
Output is correct |
17 |
Correct |
102 ms |
17696 KB |
Output is correct |
18 |
Correct |
91 ms |
18540 KB |
Output is correct |
19 |
Correct |
120 ms |
18480 KB |
Output is correct |
20 |
Correct |
272 ms |
24500 KB |
Output is correct |
21 |
Correct |
265 ms |
29080 KB |
Output is correct |
22 |
Incorrect |
290 ms |
29088 KB |
Output isn't correct |
23 |
Halted |
0 ms |
0 KB |
- |