#include "split.h"
#include <iostream>
#include <vector>
#include <array>
#include <algorithm>
#define ll long long
using namespace std;
vector <ll> V;
vector <ll> adj[100000], E[200000];
bool ap[200000], ok;
vector <ll> G[200000], X[100000];
bool visited[100000];
array<ll, 2> T[3];
ll m, k, A[3], B[3], low[100000], st[100000], sz[200000], F[200000], C[200000], cnt;
void tar(ll u, ll p) {
V.push_back(u);
low[u] = st[u] = ++cnt;
ll rch = 0;
for (auto v : adj[u]) {
if (!st[v]) {
++rch;
tar(v, u);
low[u] = min(low[u], low[v]);
if (low[v] >= st[u]) {
if (p != -1) ap[u] = 1;
while (!V.empty()) {
auto x = V.back();
V.pop_back();
X[x].push_back(k);
G[k].push_back(x);
if (x == v) break;
}
X[u].push_back(k);
G[k].push_back(u);
++k;
}
}
else if (v != p) {
low[u] = min(low[u], st[v]);
}
}
if (p == -1 & rch > 1) ap[u] = 1;
}
void dfs(ll u, ll w) {
visited[u] = 1;
for (auto v : adj[u]) {
if (!visited[v] && !C[v]) {
if (B[w] < A[w]) {
F[v] = w;
++B[w];
}
else F[v] = 2;
dfs(v, w);
}
}
}
void find(ll u, ll w) {
if (B[w] < A[w]) {
++B[w];
F[u] = w;
}
else F[u] = 2;
dfs(u, w);
for (auto v : adj[u]) {
if (C[v] && !visited[v]) {
find(v, w);
}
}
}
void solve(ll u, ll p) {
if (ap[u]) ++sz[u];
else sz[u] += G[u].size() - E[u].size();
ll mx = -1;
for (auto v : E[u]) {
if (v != p) {
solve(v, u);
sz[u] += sz[v];
mx = max(mx, sz[v]);
}
}
mx = max(mx, m - sz[u]);
if (!ok && !ap[u] && mx <= m-A[0]) {
if (mx >= A[1]) {
if (m-sz[u] == mx) {
F[p] = 1;
}
else {
for (auto v : E[u]) {
if (v != p) {
if (sz[v] == mx) {
F[v] = 1;
break;
}
}
}
}
for (auto v : G[u]) {
C[v] = 1;
}
for (auto v : G[u]) {
if (F[v]) {
++B[1];
dfs(v, 1);
}
}
for (auto v : G[u]) {
if (!F[v]) {
find(v, 0);
break;
}
}
}
else {
ll cur = 0;
for (int i=0; i<m; ++i) F[i] = 2;
for (auto v : G[u]) {
C[v] = 1;
}
find(G[u][0], 0);
for (int i=0; i<m; ++i) {
if (F[i] != 0) visited[i] = 0;
}
for (auto v : G[u]) {
if (!visited[v]) {
find(v, 1);
break;
}
}
}
ok = 1;
return;
}
}
vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
A[0] = T[0][0] = a, A[1] = T[1][0] = b, A[2] = T[2][0] = c;
T[1][1] = 1, T[2][1] = 2;
sort(A, A+3);
sort(T, T+3);
m = n;
k = n;
for (int i=0; i<p.size(); ++i) {
adj[p[i]].push_back(q[i]);
adj[q[i]].push_back(p[i]);
}
tar(0, -1);
for (int i=0; i<n; ++i) {
if (ap[i]) {
for (auto u : X[i]) {
E[i].push_back(u);
E[u].push_back(i);
}
}
}
solve(n, -1);
vector <int> U;
if (ok) {
U.clear();
for (int i=0; i<n; ++i) {
U.push_back(T[F[i]][1]+1);
}
}
else {
for (int i=0; i<n; ++i) {
U.push_back(F[i]);
}
}
return U;
}
Compilation message
split.cpp: In function 'void tar(long long int, long long int)':
split.cpp:45:8: warning: suggest parentheses around comparison in operand of '&' [-Wparentheses]
45 | if (p == -1 & rch > 1) ap[u] = 1;
| ~~^~~~~
split.cpp: In function 'void solve(long long int, long long int)':
split.cpp:120:7: warning: unused variable 'cur' [-Wunused-variable]
120 | ll cur = 0;
| ^~~
split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:148:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
148 | for (int i=0; i<p.size(); ++i) {
| ~^~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
5 ms |
18780 KB |
ok, correct split |
2 |
Correct |
4 ms |
18780 KB |
ok, correct split |
3 |
Correct |
4 ms |
18776 KB |
ok, correct split |
4 |
Correct |
6 ms |
18780 KB |
ok, correct split |
5 |
Correct |
4 ms |
18780 KB |
ok, correct split |
6 |
Correct |
5 ms |
18780 KB |
ok, correct split |
7 |
Correct |
165 ms |
54608 KB |
ok, correct split |
8 |
Correct |
142 ms |
60680 KB |
ok, correct split |
9 |
Correct |
180 ms |
60332 KB |
ok, correct split |
10 |
Correct |
76 ms |
40304 KB |
ok, correct split |
11 |
Correct |
94 ms |
40424 KB |
ok, correct split |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
18780 KB |
ok, correct split |
2 |
Correct |
4 ms |
18776 KB |
ok, correct split |
3 |
Correct |
4 ms |
18776 KB |
ok, correct split |
4 |
Correct |
108 ms |
39104 KB |
ok, correct split |
5 |
Correct |
116 ms |
38096 KB |
ok, correct split |
6 |
Correct |
74 ms |
40392 KB |
ok, correct split |
7 |
Correct |
163 ms |
60372 KB |
ok, correct split |
8 |
Correct |
111 ms |
33868 KB |
ok, correct split |
9 |
Correct |
71 ms |
28832 KB |
ok, correct split |
10 |
Correct |
79 ms |
36548 KB |
ok, correct split |
11 |
Correct |
61 ms |
36552 KB |
ok, correct split |
12 |
Correct |
63 ms |
36544 KB |
ok, correct split |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
18780 KB |
ok, correct split |
2 |
Correct |
103 ms |
38096 KB |
ok, correct split |
3 |
Correct |
36 ms |
26756 KB |
ok, correct split |
4 |
Correct |
5 ms |
18780 KB |
ok, correct split |
5 |
Correct |
154 ms |
45776 KB |
ok, correct split |
6 |
Correct |
153 ms |
45436 KB |
ok, correct split |
7 |
Correct |
131 ms |
45648 KB |
ok, correct split |
8 |
Correct |
152 ms |
45856 KB |
ok, correct split |
9 |
Correct |
141 ms |
45416 KB |
ok, correct split |
10 |
Correct |
28 ms |
23388 KB |
ok, no valid answer |
11 |
Correct |
41 ms |
26524 KB |
ok, no valid answer |
12 |
Correct |
97 ms |
35576 KB |
ok, no valid answer |
13 |
Correct |
111 ms |
35892 KB |
ok, no valid answer |
14 |
Correct |
58 ms |
34496 KB |
ok, no valid answer |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
18780 KB |
ok, correct split |
2 |
Correct |
4 ms |
16732 KB |
ok, no valid answer |
3 |
Correct |
4 ms |
18780 KB |
ok, correct split |
4 |
Correct |
4 ms |
18780 KB |
ok, correct split |
5 |
Correct |
4 ms |
18780 KB |
ok, correct split |
6 |
Correct |
5 ms |
18776 KB |
ok, correct split |
7 |
Correct |
5 ms |
18776 KB |
ok, correct split |
8 |
Correct |
5 ms |
18776 KB |
ok, correct split |
9 |
Correct |
7 ms |
19292 KB |
ok, correct split |
10 |
Correct |
7 ms |
19288 KB |
ok, correct split |
11 |
Correct |
5 ms |
18780 KB |
ok, correct split |
12 |
Incorrect |
6 ms |
19292 KB |
invalid split: #1=800, #2=1599, #3=2 |
13 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
5 ms |
18780 KB |
ok, correct split |
2 |
Correct |
4 ms |
18780 KB |
ok, correct split |
3 |
Correct |
4 ms |
18776 KB |
ok, correct split |
4 |
Correct |
6 ms |
18780 KB |
ok, correct split |
5 |
Correct |
4 ms |
18780 KB |
ok, correct split |
6 |
Correct |
5 ms |
18780 KB |
ok, correct split |
7 |
Correct |
165 ms |
54608 KB |
ok, correct split |
8 |
Correct |
142 ms |
60680 KB |
ok, correct split |
9 |
Correct |
180 ms |
60332 KB |
ok, correct split |
10 |
Correct |
76 ms |
40304 KB |
ok, correct split |
11 |
Correct |
94 ms |
40424 KB |
ok, correct split |
12 |
Correct |
4 ms |
18780 KB |
ok, correct split |
13 |
Correct |
4 ms |
18776 KB |
ok, correct split |
14 |
Correct |
4 ms |
18776 KB |
ok, correct split |
15 |
Correct |
108 ms |
39104 KB |
ok, correct split |
16 |
Correct |
116 ms |
38096 KB |
ok, correct split |
17 |
Correct |
74 ms |
40392 KB |
ok, correct split |
18 |
Correct |
163 ms |
60372 KB |
ok, correct split |
19 |
Correct |
111 ms |
33868 KB |
ok, correct split |
20 |
Correct |
71 ms |
28832 KB |
ok, correct split |
21 |
Correct |
79 ms |
36548 KB |
ok, correct split |
22 |
Correct |
61 ms |
36552 KB |
ok, correct split |
23 |
Correct |
63 ms |
36544 KB |
ok, correct split |
24 |
Correct |
4 ms |
18780 KB |
ok, correct split |
25 |
Correct |
103 ms |
38096 KB |
ok, correct split |
26 |
Correct |
36 ms |
26756 KB |
ok, correct split |
27 |
Correct |
5 ms |
18780 KB |
ok, correct split |
28 |
Correct |
154 ms |
45776 KB |
ok, correct split |
29 |
Correct |
153 ms |
45436 KB |
ok, correct split |
30 |
Correct |
131 ms |
45648 KB |
ok, correct split |
31 |
Correct |
152 ms |
45856 KB |
ok, correct split |
32 |
Correct |
141 ms |
45416 KB |
ok, correct split |
33 |
Correct |
28 ms |
23388 KB |
ok, no valid answer |
34 |
Correct |
41 ms |
26524 KB |
ok, no valid answer |
35 |
Correct |
97 ms |
35576 KB |
ok, no valid answer |
36 |
Correct |
111 ms |
35892 KB |
ok, no valid answer |
37 |
Correct |
58 ms |
34496 KB |
ok, no valid answer |
38 |
Correct |
4 ms |
18780 KB |
ok, correct split |
39 |
Correct |
4 ms |
16732 KB |
ok, no valid answer |
40 |
Correct |
4 ms |
18780 KB |
ok, correct split |
41 |
Correct |
4 ms |
18780 KB |
ok, correct split |
42 |
Correct |
4 ms |
18780 KB |
ok, correct split |
43 |
Correct |
5 ms |
18776 KB |
ok, correct split |
44 |
Correct |
5 ms |
18776 KB |
ok, correct split |
45 |
Correct |
5 ms |
18776 KB |
ok, correct split |
46 |
Correct |
7 ms |
19292 KB |
ok, correct split |
47 |
Correct |
7 ms |
19288 KB |
ok, correct split |
48 |
Correct |
5 ms |
18780 KB |
ok, correct split |
49 |
Incorrect |
6 ms |
19292 KB |
invalid split: #1=800, #2=1599, #3=2 |
50 |
Halted |
0 ms |
0 KB |
- |