# |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
223756 |
2020-04-16T10:18:34 Z |
johutha |
Railway (BOI17_railway) |
C++17 |
|
242 ms |
61036 KB |
#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
#define int int64_t
using namespace std;
int log2(int n)
{
int r = 0;
for (; n; n /= 2) r++;
return r;
}
struct tree
{
int n;
vector<vector<int>> adjlist;
vector<vector<int>> st;
vector<vector<int>> en;
vector<int> res;
set<int>* smldfs(int curr, int par)
{
set<int>* rs = new set<int>(st[curr].begin(), st[curr].end());
for (auto next : adjlist[curr])
{
if (next == par) continue;
set<int>* ot = smldfs(next, curr);
if (ot->size() > rs->size()) swap(ot, rs);
for (auto i : (*ot))
{
rs->insert(i);
}
}
for (auto i : en[curr])
{
rs->erase(i);
}
res[curr] = rs->size();
return rs;
}
vector<vector<int>> table;
vector<int> lvl;
int logn;
void dfs(int curr, int par, int lv)
{
lvl[curr] = lv;
table[0][curr] = par;
for (auto next : adjlist[curr]) if (next != par) dfs(next, curr, lv + 1);
}
void prelca()
{
logn = log2(n) + 1;
table.resize(logn, vector<int>(n));
lvl.resize(n);
dfs(0, -1, 0);
for (int i = 1; i < logn; i++)
{
for (int j = 0; j < n; j++)
{
table[i][j] = (table[i - 1][j] == -1 ? -1 : table[i - 1][table[i - 1][j]]);
}
}
}
int lca(int a, int b)
{
if (lvl[a] < lvl[b]) swap(a, b);
int ld = lvl[a] - lvl[b];
for (int j = 0; j < logn; j++)
{
if (ld & (1<<j)) a = table[j][a];
}
if (a == b) return a;
for (int j = logn - 1; j >= 0; j--)
{
if (table[j][a] != table[j][b])
{
a = table[j][a];
b = table[j][b];
}
}
return table[0][a];
}
};
signed main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
int n, m, k;
cin >> n >> m >> k;
tree t;
t.n = n;
t.adjlist.resize(n);
vector<pair<int,int>> edges;
for (int i = 0; i < n - 1; i++)
{
int a, b;
cin >> a >> b;
a--; b--;
edges.emplace_back(a, b);
t.adjlist[a].push_back(b);
t.adjlist[b].push_back(a);
}
t.prelca();
t.st.resize(n);
t.en.resize(n);
for (int i = 0; i < m; i++)
{
int c;
cin >> c;
int l = -1;
for (int j = 0; j < c; j++)
{
int a;
cin >> a;
a--;
t.st[a].push_back(i);
if (l == -1) l = a;
else l = t.lca(l, a);
}
t.en[l].push_back(i);
}
t.res.resize(n);
t.smldfs(0, -1);
vector<int> sol;
for (int i = 0; i < n - 1; i++)
{
auto e = edges[i];
int a = (t.lvl[e.first] > t.lvl[e.second] ? e.first : e.second);
if (t.res[a] >= k) sol.push_back(i + 1);
}
cout << sol.size() << "\n";
for (auto i : sol) cout << i << " ";
cout << "\n";
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
384 KB |
Output is correct |
2 |
Correct |
11 ms |
3836 KB |
Output is correct |
3 |
Correct |
11 ms |
3836 KB |
Output is correct |
4 |
Correct |
5 ms |
384 KB |
Output is correct |
5 |
Correct |
5 ms |
384 KB |
Output is correct |
6 |
Correct |
12 ms |
4348 KB |
Output is correct |
7 |
Correct |
11 ms |
3964 KB |
Output is correct |
8 |
Correct |
10 ms |
3964 KB |
Output is correct |
9 |
Correct |
11 ms |
3964 KB |
Output is correct |
10 |
Correct |
5 ms |
384 KB |
Output is correct |
11 |
Correct |
5 ms |
384 KB |
Output is correct |
12 |
Correct |
4 ms |
384 KB |
Output is correct |
13 |
Correct |
5 ms |
256 KB |
Output is correct |
14 |
Correct |
5 ms |
384 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
384 KB |
Output is correct |
2 |
Correct |
11 ms |
3836 KB |
Output is correct |
3 |
Correct |
11 ms |
3836 KB |
Output is correct |
4 |
Correct |
5 ms |
384 KB |
Output is correct |
5 |
Correct |
5 ms |
384 KB |
Output is correct |
6 |
Correct |
12 ms |
4348 KB |
Output is correct |
7 |
Correct |
11 ms |
3964 KB |
Output is correct |
8 |
Correct |
10 ms |
3964 KB |
Output is correct |
9 |
Correct |
11 ms |
3964 KB |
Output is correct |
10 |
Correct |
5 ms |
384 KB |
Output is correct |
11 |
Correct |
5 ms |
384 KB |
Output is correct |
12 |
Correct |
4 ms |
384 KB |
Output is correct |
13 |
Correct |
5 ms |
256 KB |
Output is correct |
14 |
Correct |
5 ms |
384 KB |
Output is correct |
15 |
Correct |
42 ms |
10612 KB |
Output is correct |
16 |
Correct |
44 ms |
10740 KB |
Output is correct |
17 |
Correct |
44 ms |
10740 KB |
Output is correct |
18 |
Correct |
13 ms |
4380 KB |
Output is correct |
19 |
Correct |
12 ms |
3964 KB |
Output is correct |
20 |
Correct |
45 ms |
10356 KB |
Output is correct |
21 |
Correct |
47 ms |
10104 KB |
Output is correct |
22 |
Correct |
5 ms |
384 KB |
Output is correct |
23 |
Correct |
11 ms |
3808 KB |
Output is correct |
24 |
Correct |
11 ms |
3836 KB |
Output is correct |
25 |
Correct |
4 ms |
384 KB |
Output is correct |
26 |
Correct |
4 ms |
384 KB |
Output is correct |
27 |
Correct |
12 ms |
4348 KB |
Output is correct |
28 |
Correct |
11 ms |
3836 KB |
Output is correct |
29 |
Correct |
10 ms |
3840 KB |
Output is correct |
30 |
Correct |
11 ms |
3964 KB |
Output is correct |
31 |
Correct |
5 ms |
384 KB |
Output is correct |
32 |
Correct |
4 ms |
384 KB |
Output is correct |
33 |
Correct |
4 ms |
384 KB |
Output is correct |
34 |
Correct |
5 ms |
384 KB |
Output is correct |
35 |
Correct |
5 ms |
384 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
198 ms |
53992 KB |
Output is correct |
2 |
Correct |
4 ms |
384 KB |
Output is correct |
3 |
Correct |
201 ms |
53096 KB |
Output is correct |
4 |
Correct |
192 ms |
52840 KB |
Output is correct |
5 |
Correct |
182 ms |
52840 KB |
Output is correct |
6 |
Correct |
160 ms |
55784 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
134 ms |
46440 KB |
Output is correct |
2 |
Correct |
154 ms |
46620 KB |
Output is correct |
3 |
Correct |
193 ms |
51564 KB |
Output is correct |
4 |
Correct |
183 ms |
53356 KB |
Output is correct |
5 |
Correct |
188 ms |
51820 KB |
Output is correct |
6 |
Correct |
136 ms |
47720 KB |
Output is correct |
7 |
Correct |
131 ms |
48104 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
134 ms |
46440 KB |
Output is correct |
2 |
Correct |
154 ms |
46620 KB |
Output is correct |
3 |
Correct |
193 ms |
51564 KB |
Output is correct |
4 |
Correct |
183 ms |
53356 KB |
Output is correct |
5 |
Correct |
188 ms |
51820 KB |
Output is correct |
6 |
Correct |
136 ms |
47720 KB |
Output is correct |
7 |
Correct |
131 ms |
48104 KB |
Output is correct |
8 |
Correct |
145 ms |
47976 KB |
Output is correct |
9 |
Correct |
146 ms |
47848 KB |
Output is correct |
10 |
Correct |
148 ms |
52712 KB |
Output is correct |
11 |
Correct |
144 ms |
52584 KB |
Output is correct |
12 |
Correct |
112 ms |
43112 KB |
Output is correct |
13 |
Correct |
109 ms |
43112 KB |
Output is correct |
14 |
Correct |
158 ms |
48212 KB |
Output is correct |
15 |
Correct |
166 ms |
48500 KB |
Output is correct |
16 |
Correct |
188 ms |
54252 KB |
Output is correct |
17 |
Correct |
186 ms |
52464 KB |
Output is correct |
18 |
Correct |
179 ms |
52724 KB |
Output is correct |
19 |
Correct |
159 ms |
47464 KB |
Output is correct |
20 |
Correct |
139 ms |
48360 KB |
Output is correct |
21 |
Correct |
139 ms |
48872 KB |
Output is correct |
22 |
Correct |
137 ms |
49000 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
384 KB |
Output is correct |
2 |
Correct |
11 ms |
3836 KB |
Output is correct |
3 |
Correct |
11 ms |
3836 KB |
Output is correct |
4 |
Correct |
5 ms |
384 KB |
Output is correct |
5 |
Correct |
5 ms |
384 KB |
Output is correct |
6 |
Correct |
12 ms |
4348 KB |
Output is correct |
7 |
Correct |
11 ms |
3964 KB |
Output is correct |
8 |
Correct |
10 ms |
3964 KB |
Output is correct |
9 |
Correct |
11 ms |
3964 KB |
Output is correct |
10 |
Correct |
5 ms |
384 KB |
Output is correct |
11 |
Correct |
5 ms |
384 KB |
Output is correct |
12 |
Correct |
4 ms |
384 KB |
Output is correct |
13 |
Correct |
5 ms |
256 KB |
Output is correct |
14 |
Correct |
5 ms |
384 KB |
Output is correct |
15 |
Correct |
42 ms |
10612 KB |
Output is correct |
16 |
Correct |
44 ms |
10740 KB |
Output is correct |
17 |
Correct |
44 ms |
10740 KB |
Output is correct |
18 |
Correct |
13 ms |
4380 KB |
Output is correct |
19 |
Correct |
12 ms |
3964 KB |
Output is correct |
20 |
Correct |
45 ms |
10356 KB |
Output is correct |
21 |
Correct |
47 ms |
10104 KB |
Output is correct |
22 |
Correct |
5 ms |
384 KB |
Output is correct |
23 |
Correct |
11 ms |
3808 KB |
Output is correct |
24 |
Correct |
11 ms |
3836 KB |
Output is correct |
25 |
Correct |
4 ms |
384 KB |
Output is correct |
26 |
Correct |
4 ms |
384 KB |
Output is correct |
27 |
Correct |
12 ms |
4348 KB |
Output is correct |
28 |
Correct |
11 ms |
3836 KB |
Output is correct |
29 |
Correct |
10 ms |
3840 KB |
Output is correct |
30 |
Correct |
11 ms |
3964 KB |
Output is correct |
31 |
Correct |
5 ms |
384 KB |
Output is correct |
32 |
Correct |
4 ms |
384 KB |
Output is correct |
33 |
Correct |
4 ms |
384 KB |
Output is correct |
34 |
Correct |
5 ms |
384 KB |
Output is correct |
35 |
Correct |
5 ms |
384 KB |
Output is correct |
36 |
Correct |
198 ms |
53992 KB |
Output is correct |
37 |
Correct |
4 ms |
384 KB |
Output is correct |
38 |
Correct |
201 ms |
53096 KB |
Output is correct |
39 |
Correct |
192 ms |
52840 KB |
Output is correct |
40 |
Correct |
182 ms |
52840 KB |
Output is correct |
41 |
Correct |
160 ms |
55784 KB |
Output is correct |
42 |
Correct |
134 ms |
46440 KB |
Output is correct |
43 |
Correct |
154 ms |
46620 KB |
Output is correct |
44 |
Correct |
193 ms |
51564 KB |
Output is correct |
45 |
Correct |
183 ms |
53356 KB |
Output is correct |
46 |
Correct |
188 ms |
51820 KB |
Output is correct |
47 |
Correct |
136 ms |
47720 KB |
Output is correct |
48 |
Correct |
131 ms |
48104 KB |
Output is correct |
49 |
Correct |
145 ms |
47976 KB |
Output is correct |
50 |
Correct |
146 ms |
47848 KB |
Output is correct |
51 |
Correct |
148 ms |
52712 KB |
Output is correct |
52 |
Correct |
144 ms |
52584 KB |
Output is correct |
53 |
Correct |
112 ms |
43112 KB |
Output is correct |
54 |
Correct |
109 ms |
43112 KB |
Output is correct |
55 |
Correct |
158 ms |
48212 KB |
Output is correct |
56 |
Correct |
166 ms |
48500 KB |
Output is correct |
57 |
Correct |
188 ms |
54252 KB |
Output is correct |
58 |
Correct |
186 ms |
52464 KB |
Output is correct |
59 |
Correct |
179 ms |
52724 KB |
Output is correct |
60 |
Correct |
159 ms |
47464 KB |
Output is correct |
61 |
Correct |
139 ms |
48360 KB |
Output is correct |
62 |
Correct |
139 ms |
48872 KB |
Output is correct |
63 |
Correct |
137 ms |
49000 KB |
Output is correct |
64 |
Correct |
153 ms |
48232 KB |
Output is correct |
65 |
Correct |
242 ms |
61036 KB |
Output is correct |
66 |
Correct |
195 ms |
56560 KB |
Output is correct |
67 |
Correct |
178 ms |
51948 KB |
Output is correct |
68 |
Correct |
95 ms |
40812 KB |
Output is correct |
69 |
Correct |
94 ms |
40812 KB |
Output is correct |
70 |
Correct |
157 ms |
49512 KB |
Output is correct |
71 |
Correct |
146 ms |
47464 KB |
Output is correct |
72 |
Correct |
5 ms |
384 KB |
Output is correct |
73 |
Correct |
11 ms |
3836 KB |
Output is correct |
74 |
Correct |
11 ms |
3836 KB |
Output is correct |
75 |
Correct |
5 ms |
384 KB |
Output is correct |
76 |
Correct |
4 ms |
384 KB |
Output is correct |
77 |
Correct |
12 ms |
4348 KB |
Output is correct |
78 |
Correct |
10 ms |
3836 KB |
Output is correct |
79 |
Correct |
10 ms |
3964 KB |
Output is correct |
80 |
Correct |
10 ms |
3964 KB |
Output is correct |
81 |
Correct |
4 ms |
384 KB |
Output is correct |
82 |
Correct |
4 ms |
384 KB |
Output is correct |
83 |
Correct |
5 ms |
384 KB |
Output is correct |
84 |
Correct |
4 ms |
384 KB |
Output is correct |
85 |
Correct |
4 ms |
384 KB |
Output is correct |
86 |
Correct |
43 ms |
10612 KB |
Output is correct |
87 |
Correct |
44 ms |
10740 KB |
Output is correct |
88 |
Correct |
43 ms |
10740 KB |
Output is correct |
89 |
Correct |
12 ms |
4476 KB |
Output is correct |
90 |
Correct |
11 ms |
3964 KB |
Output is correct |
91 |
Correct |
45 ms |
10356 KB |
Output is correct |
92 |
Correct |
45 ms |
10232 KB |
Output is correct |
93 |
Correct |
4 ms |
384 KB |
Output is correct |
94 |
Correct |
198 ms |
54140 KB |
Output is correct |
95 |
Correct |
193 ms |
53120 KB |
Output is correct |
96 |
Correct |
190 ms |
52840 KB |
Output is correct |
97 |
Correct |
203 ms |
52824 KB |
Output is correct |
98 |
Correct |
156 ms |
55784 KB |
Output is correct |
99 |
Correct |
203 ms |
57068 KB |
Output is correct |
100 |
Correct |
196 ms |
53964 KB |
Output is correct |
101 |
Correct |
187 ms |
54636 KB |
Output is correct |
102 |
Correct |
163 ms |
47208 KB |
Output is correct |
103 |
Correct |
136 ms |
47720 KB |
Output is correct |
104 |
Correct |
142 ms |
49000 KB |
Output is correct |
105 |
Correct |
132 ms |
47592 KB |
Output is correct |
106 |
Correct |
147 ms |
48104 KB |
Output is correct |
107 |
Correct |
161 ms |
48232 KB |
Output is correct |
108 |
Correct |
147 ms |
52712 KB |
Output is correct |
109 |
Correct |
144 ms |
52712 KB |
Output is correct |
110 |
Correct |
104 ms |
43112 KB |
Output is correct |
111 |
Correct |
115 ms |
43112 KB |
Output is correct |
112 |
Correct |
151 ms |
48108 KB |
Output is correct |
113 |
Correct |
149 ms |
47980 KB |
Output is correct |