# |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
826915 |
2023-08-16T06:35:08 Z |
ymm |
Keys (IOI21_keys) |
C++17 |
|
2552 ms |
158856 KB |
#include <bits/stdc++.h>
#define Loop(x,l,r) for (ll x = (l); x < (r); ++x)
typedef long long ll;
typedef std::pair<int,int> pii;
using namespace std;
const int N = 300'010;
vector<pii> A[N];
int key[N];
template<class T>
void clear_and_free(T &x) { T tmp; tmp.swap(x); }
namespace dsu {
struct cmp_t {
int vis;
cmp_t *par;
vector<int> nds;
vector<int> adj;
map<int,vector<int>> if_had_key;
set<int> keys;
bool sink;
int cost;
} cmp[N];
void init(int n)
{
Loop (i,0,n) {
auto *c = &cmp[i];
c->vis = 0;
c->par = 0;
c->nds = {(int)i};
for (auto [v, k] : A[i]) {
if (k == key[i])
c->adj.push_back(v);
else
c->if_had_key[k].push_back(v);
}
c->keys = {key[i]};
c->sink = 0;
c->cost = 5 + A[i].size();
}
}
cmp_t *rt(cmp_t *c) { return c->par? (c->par = rt(c->par)): c; }
cmp_t *rt(int v) { return rt(&cmp[v]); }
void merge(cmp_t *c, cmp_t *d) {
c = rt(c);
d = rt(d);
if (c == d)
return;
//cerr << "merge(" << c-cmp << ", " << d-cmp << ")\n";
if (c->cost < d->cost)
swap(c, d);
c->cost += d->cost;
c->nds.insert(c->nds.end(), d->nds.begin(), d->nds.end());
c->adj.insert(c->adj.end(), d->adj.begin(), d->adj.end());
for (auto k : d->keys)
c->cost -= 4*!c->keys.insert(k).second;
for (auto &[k, vec] : d->if_had_key) {
if (c->keys.count(k)) {
c->adj.insert(c->adj.end(), vec.begin(), vec.end());
} else {
auto &tmp = c->if_had_key[k];
tmp.insert(tmp.end(), vec.begin(), vec.end());
}
}
for (auto k : d->keys) {
auto &vec = c->if_had_key[k];
c->adj.insert(c->adj.end(), vec.begin(), vec.end());
clear_and_free(vec);
}
d->par = c;
clear_and_free(d->nds);
clear_and_free(d->adj);
clear_and_free(d->if_had_key);
clear_and_free(d->keys);
}
}
typedef dsu::cmp_t *cmp;
void dfs(cmp v, int t)
{
vector<cmp> anc;
anc.push_back(v);
v->vis = t;
for (;;) {
cmp v = anc.back();
//cerr << v-dsu::cmp << '\n';
if (v->adj.size()) {
auto u = dsu::rt(v->adj.back());
v->adj.pop_back();
v->cost--;
if (u == v)
continue;
cerr << v-dsu::cmp << " -> " << u-dsu::cmp << '\n';
if (u->vis == t) {
while (dsu::rt(v) != dsu::rt(u)) {
auto x = anc.back(); anc.pop_back();
auto y = anc.back(); anc.pop_back();
dsu::merge(x, y);
anc.push_back(dsu::rt(x));
}
} else if (u->vis) {
return;
} else {
u->vis = t;
anc.push_back(u);
}
} else {
v->sink = 1;
return;
}
}
}
std::vector<int> find_reachable(std::vector<int> r, std::vector<int> u, std::vector<int> v, std::vector<int> c) {
int n = r.size();
int m = u.size();
Loop (i,0,n)
key[i] = r[i];
Loop (i,0,m) {
int x = v[i], y = u[i], z = c[i];
A[x].emplace_back(y, z);
A[y].emplace_back(x, z);
}
dsu::init(n);
int nxt = 1;
Loop (i,0,n) {
cmp c = &dsu::cmp[i];
if (c->par || c->vis)
continue;
dfs(c, nxt++);
}
int mn = N;
Loop (i,0,n) {
cmp c = &dsu::cmp[i];
if (c->sink) {
//cerr << c-dsu::cmp << ": " << c->nds.size() << '\n';
mn = min<int>(mn, c->nds.size());
}
}
vector<int> ans(n);
Loop (i,0,n) {
cmp c = &dsu::cmp[i];
if (c->sink && c->nds.size() == mn) {
for (auto x : c->nds)
ans[x] = 1;
}
}
return ans;
}
Compilation message
keys.cpp: In function 'std::vector<int> find_reachable(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
keys.cpp:146:32: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
146 | if (c->sink && c->nds.size() == mn) {
| ~~~~~~~~~~~~~~^~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
23 ms |
56660 KB |
Output is correct |
2 |
Correct |
23 ms |
56568 KB |
Output is correct |
3 |
Correct |
22 ms |
56604 KB |
Output is correct |
4 |
Correct |
24 ms |
56660 KB |
Output is correct |
5 |
Correct |
24 ms |
56676 KB |
Output is correct |
6 |
Correct |
23 ms |
56660 KB |
Output is correct |
7 |
Correct |
23 ms |
56660 KB |
Output is correct |
8 |
Correct |
25 ms |
56644 KB |
Output is correct |
9 |
Correct |
24 ms |
56640 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
23 ms |
56660 KB |
Output is correct |
2 |
Correct |
23 ms |
56568 KB |
Output is correct |
3 |
Correct |
22 ms |
56604 KB |
Output is correct |
4 |
Correct |
24 ms |
56660 KB |
Output is correct |
5 |
Correct |
24 ms |
56676 KB |
Output is correct |
6 |
Correct |
23 ms |
56660 KB |
Output is correct |
7 |
Correct |
23 ms |
56660 KB |
Output is correct |
8 |
Correct |
25 ms |
56644 KB |
Output is correct |
9 |
Correct |
24 ms |
56640 KB |
Output is correct |
10 |
Correct |
23 ms |
56660 KB |
Output is correct |
11 |
Correct |
23 ms |
56672 KB |
Output is correct |
12 |
Correct |
24 ms |
56604 KB |
Output is correct |
13 |
Correct |
23 ms |
56668 KB |
Output is correct |
14 |
Correct |
23 ms |
56660 KB |
Output is correct |
15 |
Correct |
25 ms |
56660 KB |
Output is correct |
16 |
Correct |
24 ms |
56672 KB |
Output is correct |
17 |
Correct |
24 ms |
56612 KB |
Output is correct |
18 |
Correct |
24 ms |
56636 KB |
Output is correct |
19 |
Correct |
24 ms |
56676 KB |
Output is correct |
20 |
Correct |
24 ms |
56660 KB |
Output is correct |
21 |
Correct |
24 ms |
56692 KB |
Output is correct |
22 |
Correct |
24 ms |
56676 KB |
Output is correct |
23 |
Correct |
25 ms |
56788 KB |
Output is correct |
24 |
Correct |
24 ms |
56660 KB |
Output is correct |
25 |
Correct |
25 ms |
56668 KB |
Output is correct |
26 |
Correct |
25 ms |
56648 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
23 ms |
56660 KB |
Output is correct |
2 |
Correct |
23 ms |
56568 KB |
Output is correct |
3 |
Correct |
22 ms |
56604 KB |
Output is correct |
4 |
Correct |
24 ms |
56660 KB |
Output is correct |
5 |
Correct |
24 ms |
56676 KB |
Output is correct |
6 |
Correct |
23 ms |
56660 KB |
Output is correct |
7 |
Correct |
23 ms |
56660 KB |
Output is correct |
8 |
Correct |
25 ms |
56644 KB |
Output is correct |
9 |
Correct |
24 ms |
56640 KB |
Output is correct |
10 |
Correct |
23 ms |
56660 KB |
Output is correct |
11 |
Correct |
23 ms |
56672 KB |
Output is correct |
12 |
Correct |
24 ms |
56604 KB |
Output is correct |
13 |
Correct |
23 ms |
56668 KB |
Output is correct |
14 |
Correct |
23 ms |
56660 KB |
Output is correct |
15 |
Correct |
25 ms |
56660 KB |
Output is correct |
16 |
Correct |
24 ms |
56672 KB |
Output is correct |
17 |
Correct |
24 ms |
56612 KB |
Output is correct |
18 |
Correct |
24 ms |
56636 KB |
Output is correct |
19 |
Correct |
24 ms |
56676 KB |
Output is correct |
20 |
Correct |
24 ms |
56660 KB |
Output is correct |
21 |
Correct |
24 ms |
56692 KB |
Output is correct |
22 |
Correct |
24 ms |
56676 KB |
Output is correct |
23 |
Correct |
25 ms |
56788 KB |
Output is correct |
24 |
Correct |
24 ms |
56660 KB |
Output is correct |
25 |
Correct |
25 ms |
56668 KB |
Output is correct |
26 |
Correct |
25 ms |
56648 KB |
Output is correct |
27 |
Correct |
29 ms |
57284 KB |
Output is correct |
28 |
Correct |
29 ms |
57248 KB |
Output is correct |
29 |
Correct |
29 ms |
57312 KB |
Output is correct |
30 |
Correct |
28 ms |
56908 KB |
Output is correct |
31 |
Correct |
24 ms |
56908 KB |
Output is correct |
32 |
Correct |
25 ms |
56676 KB |
Output is correct |
33 |
Correct |
27 ms |
56944 KB |
Output is correct |
34 |
Correct |
29 ms |
56888 KB |
Output is correct |
35 |
Correct |
28 ms |
56900 KB |
Output is correct |
36 |
Correct |
37 ms |
57292 KB |
Output is correct |
37 |
Correct |
32 ms |
57172 KB |
Output is correct |
38 |
Correct |
34 ms |
57256 KB |
Output is correct |
39 |
Correct |
33 ms |
57232 KB |
Output is correct |
40 |
Correct |
29 ms |
56988 KB |
Output is correct |
41 |
Correct |
30 ms |
57172 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
23 ms |
56660 KB |
Output is correct |
2 |
Correct |
23 ms |
56568 KB |
Output is correct |
3 |
Correct |
22 ms |
56604 KB |
Output is correct |
4 |
Correct |
24 ms |
56660 KB |
Output is correct |
5 |
Correct |
24 ms |
56676 KB |
Output is correct |
6 |
Correct |
23 ms |
56660 KB |
Output is correct |
7 |
Correct |
23 ms |
56660 KB |
Output is correct |
8 |
Correct |
25 ms |
56644 KB |
Output is correct |
9 |
Correct |
24 ms |
56640 KB |
Output is correct |
10 |
Correct |
887 ms |
130836 KB |
Output is correct |
11 |
Correct |
2226 ms |
122592 KB |
Output is correct |
12 |
Correct |
301 ms |
74284 KB |
Output is correct |
13 |
Correct |
1509 ms |
145964 KB |
Output is correct |
14 |
Correct |
1360 ms |
124564 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
23 ms |
56660 KB |
Output is correct |
2 |
Correct |
23 ms |
56568 KB |
Output is correct |
3 |
Correct |
22 ms |
56604 KB |
Output is correct |
4 |
Correct |
24 ms |
56660 KB |
Output is correct |
5 |
Correct |
24 ms |
56676 KB |
Output is correct |
6 |
Correct |
23 ms |
56660 KB |
Output is correct |
7 |
Correct |
23 ms |
56660 KB |
Output is correct |
8 |
Correct |
25 ms |
56644 KB |
Output is correct |
9 |
Correct |
24 ms |
56640 KB |
Output is correct |
10 |
Correct |
23 ms |
56660 KB |
Output is correct |
11 |
Correct |
23 ms |
56672 KB |
Output is correct |
12 |
Correct |
24 ms |
56604 KB |
Output is correct |
13 |
Correct |
23 ms |
56668 KB |
Output is correct |
14 |
Correct |
23 ms |
56660 KB |
Output is correct |
15 |
Correct |
25 ms |
56660 KB |
Output is correct |
16 |
Correct |
24 ms |
56672 KB |
Output is correct |
17 |
Correct |
24 ms |
56612 KB |
Output is correct |
18 |
Correct |
24 ms |
56636 KB |
Output is correct |
19 |
Correct |
24 ms |
56676 KB |
Output is correct |
20 |
Correct |
24 ms |
56660 KB |
Output is correct |
21 |
Correct |
24 ms |
56692 KB |
Output is correct |
22 |
Correct |
24 ms |
56676 KB |
Output is correct |
23 |
Correct |
25 ms |
56788 KB |
Output is correct |
24 |
Correct |
24 ms |
56660 KB |
Output is correct |
25 |
Correct |
25 ms |
56668 KB |
Output is correct |
26 |
Correct |
25 ms |
56648 KB |
Output is correct |
27 |
Correct |
29 ms |
57284 KB |
Output is correct |
28 |
Correct |
29 ms |
57248 KB |
Output is correct |
29 |
Correct |
29 ms |
57312 KB |
Output is correct |
30 |
Correct |
28 ms |
56908 KB |
Output is correct |
31 |
Correct |
24 ms |
56908 KB |
Output is correct |
32 |
Correct |
25 ms |
56676 KB |
Output is correct |
33 |
Correct |
27 ms |
56944 KB |
Output is correct |
34 |
Correct |
29 ms |
56888 KB |
Output is correct |
35 |
Correct |
28 ms |
56900 KB |
Output is correct |
36 |
Correct |
37 ms |
57292 KB |
Output is correct |
37 |
Correct |
32 ms |
57172 KB |
Output is correct |
38 |
Correct |
34 ms |
57256 KB |
Output is correct |
39 |
Correct |
33 ms |
57232 KB |
Output is correct |
40 |
Correct |
29 ms |
56988 KB |
Output is correct |
41 |
Correct |
30 ms |
57172 KB |
Output is correct |
42 |
Correct |
887 ms |
130836 KB |
Output is correct |
43 |
Correct |
2226 ms |
122592 KB |
Output is correct |
44 |
Correct |
301 ms |
74284 KB |
Output is correct |
45 |
Correct |
1509 ms |
145964 KB |
Output is correct |
46 |
Correct |
1360 ms |
124564 KB |
Output is correct |
47 |
Correct |
23 ms |
56660 KB |
Output is correct |
48 |
Correct |
24 ms |
56660 KB |
Output is correct |
49 |
Correct |
24 ms |
56628 KB |
Output is correct |
50 |
Correct |
2486 ms |
126940 KB |
Output is correct |
51 |
Correct |
2552 ms |
124952 KB |
Output is correct |
52 |
Correct |
615 ms |
130672 KB |
Output is correct |
53 |
Correct |
611 ms |
130720 KB |
Output is correct |
54 |
Correct |
605 ms |
130836 KB |
Output is correct |
55 |
Correct |
1014 ms |
144524 KB |
Output is correct |
56 |
Correct |
728 ms |
151128 KB |
Output is correct |
57 |
Correct |
258 ms |
145632 KB |
Output is correct |
58 |
Correct |
1576 ms |
158856 KB |
Output is correct |
59 |
Correct |
1685 ms |
154084 KB |
Output is correct |
60 |
Correct |
1062 ms |
143448 KB |
Output is correct |
61 |
Correct |
1618 ms |
149120 KB |
Output is correct |
62 |
Correct |
1846 ms |
143416 KB |
Output is correct |
63 |
Correct |
1442 ms |
140948 KB |
Output is correct |
64 |
Correct |
50 ms |
57824 KB |
Output is correct |
65 |
Correct |
47 ms |
57804 KB |
Output is correct |
66 |
Correct |
1841 ms |
142836 KB |
Output is correct |
67 |
Correct |
155 ms |
63568 KB |
Output is correct |
68 |
Correct |
250 ms |
68308 KB |
Output is correct |
69 |
Correct |
1723 ms |
154176 KB |
Output is correct |
70 |
Correct |
523 ms |
79972 KB |
Output is correct |
71 |
Correct |
1344 ms |
127412 KB |
Output is correct |
72 |
Correct |
1670 ms |
154096 KB |
Output is correct |
73 |
Correct |
1872 ms |
142784 KB |
Output is correct |