답안 #758791

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
758791	2023-06-15T10:08:46 Z	Hacv16	Railway (BOI17_railway)	C++17	100 / 100	96 ms	45632 KB

railway

#include <bits/stdc++.h>
using namespace std;

#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt") 

#define int long long int

const int LOG = 22;
const int MAX = 5e5 + 10;
const int INF = 0x3f3f3f3f;
 
int n, m, k, f[MAX], dp[MAX][LOG], h[MAX], tin[MAX], a[MAX], timer;
vector<pair<int, int>> adj[MAX];
vector<int> ids;

void dfs(int u, int p){
	tin[u] = ++timer;
	dp[u][0] = p; h[u] = h[p] + 1;

   	for(int i = 1; i < LOG; i++)
        dp[u][i] = dp[ dp[u][i - 1] ][i - 1];

	for(auto [v, id] : adj[u]){
		if(v == p) continue;
		dfs(v, u);
	}
}

int lca(int u, int v){
	if(h[u] < h[v]) swap(u, v);
	int jump = h[u] - h[v];
 
	for(int j = LOG - 1; j >= 0; j--)
		if(jump & (1 << j)) u = dp[u][j];
 
	if(u == v) return v;
 
	for(int j = LOG - 1; j >= 0; j--)
		if(dp[u][j] != dp[v][j]) u = dp[u][j], v = dp[v][j];
 
	return dp[u][0];
}

void calcResp(int u, int p, int id){
    for(auto [v, nx] : adj[u]){
        if(v == p) continue;
        calcResp(v, u, nx);
        f[u] += f[v];
    }

    if(f[u] >= k * 2 && id != 0)
    	ids.push_back(id);
}

int32_t main(void){
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);

	cin >> n >> m >> k;

	for(int i = 1; i < n; i++){
		int u, v; cin >> u >> v;

		adj[u].emplace_back(v, i);
		adj[v].emplace_back(u, i);
	}
	
	dfs(1, 0);
    
	for(int i = 1; i <= m; i++){
        int n; cin >> n;

      	for(int i = 1; i <= n; i++)
           cin >> a[i];

        sort(a + 1, a + n + 1, [&](int i, int j){
        	return tin[i] < tin[j];
        });

        a[n + 1] = a[1];

        for(int i = 1; i <= n; i++) {
            f[a[i]]++; f[a[i + 1]]++;
           	f[lca(a[i], a[i + 1])] -= 2;
        }
    }

	calcResp(1, 0, 0);

	cout << ids.size() << '\n';

	sort(ids.begin(), ids.end());
	for(auto x : ids) cout << x << ' ';
}

Subtask #1
8.0 / 8.0

#	결과	실행 시간	메모리	Grader output
1	Correct	6 ms	11988 KB	Output is correct
2	Correct	10 ms	14576 KB	Output is correct
3	Correct	10 ms	14548 KB	Output is correct
4	Correct	6 ms	11988 KB	Output is correct
5	Correct	6 ms	11988 KB	Output is correct
6	Correct	14 ms	15060 KB	Output is correct
7	Correct	10 ms	14672 KB	Output is correct
8	Correct	9 ms	14440 KB	Output is correct
9	Correct	9 ms	14492 KB	Output is correct
10	Correct	5 ms	11988 KB	Output is correct
11	Correct	6 ms	11988 KB	Output is correct
12	Correct	6 ms	11988 KB	Output is correct
13	Correct	6 ms	11988 KB	Output is correct
14	Correct	6 ms	11988 KB	Output is correct

Subtask #2
15.0 / 15.0

#	결과	실행 시간	메모리	Grader output
1	Correct	6 ms	11988 KB	Output is correct
2	Correct	10 ms	14576 KB	Output is correct
3	Correct	10 ms	14548 KB	Output is correct
4	Correct	6 ms	11988 KB	Output is correct
5	Correct	6 ms	11988 KB	Output is correct
6	Correct	14 ms	15060 KB	Output is correct
7	Correct	10 ms	14672 KB	Output is correct
8	Correct	9 ms	14440 KB	Output is correct
9	Correct	9 ms	14492 KB	Output is correct
10	Correct	5 ms	11988 KB	Output is correct
11	Correct	6 ms	11988 KB	Output is correct
12	Correct	6 ms	11988 KB	Output is correct
13	Correct	6 ms	11988 KB	Output is correct
14	Correct	6 ms	11988 KB	Output is correct
15	Correct	28 ms	14668 KB	Output is correct
16	Correct	21 ms	14560 KB	Output is correct
17	Correct	22 ms	14708 KB	Output is correct
18	Correct	10 ms	14988 KB	Output is correct
19	Correct	11 ms	14668 KB	Output is correct
20	Correct	20 ms	14676 KB	Output is correct
21	Correct	23 ms	14420 KB	Output is correct
22	Correct	6 ms	11988 KB	Output is correct
23	Correct	10 ms	14548 KB	Output is correct
24	Correct	10 ms	14548 KB	Output is correct
25	Correct	6 ms	11988 KB	Output is correct
26	Correct	6 ms	11988 KB	Output is correct
27	Correct	11 ms	15060 KB	Output is correct
28	Correct	10 ms	14552 KB	Output is correct
29	Correct	9 ms	14564 KB	Output is correct
30	Correct	9 ms	14548 KB	Output is correct
31	Correct	5 ms	11988 KB	Output is correct
32	Correct	6 ms	11988 KB	Output is correct
33	Correct	6 ms	11988 KB	Output is correct
34	Correct	6 ms	11988 KB	Output is correct
35	Correct	7 ms	11988 KB	Output is correct

Subtask #3
7.0 / 7.0

#	결과	실행 시간	메모리	Grader output
1	Correct	93 ms	45368 KB	Output is correct
2	Correct	5 ms	11988 KB	Output is correct
3	Correct	89 ms	44824 KB	Output is correct
4	Correct	84 ms	44060 KB	Output is correct
5	Correct	96 ms	45372 KB	Output is correct
6	Correct	91 ms	45596 KB	Output is correct

Subtask #4
29.0 / 29.0

#	결과	실행 시간	메모리	Grader output
1	Correct	78 ms	40568 KB	Output is correct
2	Correct	72 ms	37120 KB	Output is correct
3	Correct	75 ms	36740 KB	Output is correct
4	Correct	77 ms	36704 KB	Output is correct
5	Correct	78 ms	36700 KB	Output is correct
6	Correct	71 ms	40908 KB	Output is correct
7	Correct	70 ms	40724 KB	Output is correct

Subtask #5
16.0 / 16.0

#	결과	실행 시간	메모리	Grader output
1	Correct	78 ms	40568 KB	Output is correct
2	Correct	72 ms	37120 KB	Output is correct
3	Correct	75 ms	36740 KB	Output is correct
4	Correct	77 ms	36704 KB	Output is correct
5	Correct	78 ms	36700 KB	Output is correct
6	Correct	71 ms	40908 KB	Output is correct
7	Correct	70 ms	40724 KB	Output is correct
8	Correct	83 ms	40908 KB	Output is correct
9	Correct	83 ms	40896 KB	Output is correct
10	Correct	79 ms	45528 KB	Output is correct
11	Correct	81 ms	45504 KB	Output is correct
12	Correct	71 ms	36452 KB	Output is correct
13	Correct	62 ms	36432 KB	Output is correct
14	Correct	82 ms	36768 KB	Output is correct
15	Correct	77 ms	36684 KB	Output is correct
16	Correct	77 ms	36748 KB	Output is correct
17	Correct	80 ms	36744 KB	Output is correct
18	Correct	78 ms	36716 KB	Output is correct
19	Correct	81 ms	37316 KB	Output is correct
20	Correct	72 ms	41092 KB	Output is correct
21	Correct	73 ms	41172 KB	Output is correct
22	Correct	71 ms	41108 KB	Output is correct

Subtask #6
25.0 / 25.0

#	결과	실행 시간	메모리	Grader output
1	Correct	6 ms	11988 KB	Output is correct
2	Correct	10 ms	14576 KB	Output is correct
3	Correct	10 ms	14548 KB	Output is correct
4	Correct	6 ms	11988 KB	Output is correct
5	Correct	6 ms	11988 KB	Output is correct
6	Correct	14 ms	15060 KB	Output is correct
7	Correct	10 ms	14672 KB	Output is correct
8	Correct	9 ms	14440 KB	Output is correct
9	Correct	9 ms	14492 KB	Output is correct
10	Correct	5 ms	11988 KB	Output is correct
11	Correct	6 ms	11988 KB	Output is correct
12	Correct	6 ms	11988 KB	Output is correct
13	Correct	6 ms	11988 KB	Output is correct
14	Correct	6 ms	11988 KB	Output is correct
15	Correct	28 ms	14668 KB	Output is correct
16	Correct	21 ms	14560 KB	Output is correct
17	Correct	22 ms	14708 KB	Output is correct
18	Correct	10 ms	14988 KB	Output is correct
19	Correct	11 ms	14668 KB	Output is correct
20	Correct	20 ms	14676 KB	Output is correct
21	Correct	23 ms	14420 KB	Output is correct
22	Correct	6 ms	11988 KB	Output is correct
23	Correct	10 ms	14548 KB	Output is correct
24	Correct	10 ms	14548 KB	Output is correct
25	Correct	6 ms	11988 KB	Output is correct
26	Correct	6 ms	11988 KB	Output is correct
27	Correct	11 ms	15060 KB	Output is correct
28	Correct	10 ms	14552 KB	Output is correct
29	Correct	9 ms	14564 KB	Output is correct
30	Correct	9 ms	14548 KB	Output is correct
31	Correct	5 ms	11988 KB	Output is correct
32	Correct	6 ms	11988 KB	Output is correct
33	Correct	6 ms	11988 KB	Output is correct
34	Correct	6 ms	11988 KB	Output is correct
35	Correct	7 ms	11988 KB	Output is correct
36	Correct	93 ms	45368 KB	Output is correct
37	Correct	5 ms	11988 KB	Output is correct
38	Correct	89 ms	44824 KB	Output is correct
39	Correct	84 ms	44060 KB	Output is correct
40	Correct	96 ms	45372 KB	Output is correct
41	Correct	91 ms	45596 KB	Output is correct
42	Correct	78 ms	40568 KB	Output is correct
43	Correct	72 ms	37120 KB	Output is correct
44	Correct	75 ms	36740 KB	Output is correct
45	Correct	77 ms	36704 KB	Output is correct
46	Correct	78 ms	36700 KB	Output is correct
47	Correct	71 ms	40908 KB	Output is correct
48	Correct	70 ms	40724 KB	Output is correct
49	Correct	83 ms	40908 KB	Output is correct
50	Correct	83 ms	40896 KB	Output is correct
51	Correct	79 ms	45528 KB	Output is correct
52	Correct	81 ms	45504 KB	Output is correct
53	Correct	71 ms	36452 KB	Output is correct
54	Correct	62 ms	36432 KB	Output is correct
55	Correct	82 ms	36768 KB	Output is correct
56	Correct	77 ms	36684 KB	Output is correct
57	Correct	77 ms	36748 KB	Output is correct
58	Correct	80 ms	36744 KB	Output is correct
59	Correct	78 ms	36716 KB	Output is correct
60	Correct	81 ms	37316 KB	Output is correct
61	Correct	72 ms	41092 KB	Output is correct
62	Correct	73 ms	41172 KB	Output is correct
63	Correct	71 ms	41108 KB	Output is correct
64	Correct	80 ms	40996 KB	Output is correct
65	Correct	83 ms	36764 KB	Output is correct
66	Correct	87 ms	36760 KB	Output is correct
67	Correct	84 ms	36788 KB	Output is correct
68	Correct	57 ms	36468 KB	Output is correct
69	Correct	58 ms	36380 KB	Output is correct
70	Correct	82 ms	41340 KB	Output is correct
71	Correct	82 ms	40656 KB	Output is correct
72	Correct	6 ms	11988 KB	Output is correct
73	Correct	10 ms	14480 KB	Output is correct
74	Correct	10 ms	14548 KB	Output is correct
75	Correct	6 ms	11988 KB	Output is correct
76	Correct	7 ms	11988 KB	Output is correct
77	Correct	14 ms	14992 KB	Output is correct
78	Correct	11 ms	14680 KB	Output is correct
79	Correct	10 ms	14548 KB	Output is correct
80	Correct	9 ms	14548 KB	Output is correct
81	Correct	6 ms	11988 KB	Output is correct
82	Correct	6 ms	11988 KB	Output is correct
83	Correct	6 ms	11988 KB	Output is correct
84	Correct	6 ms	12104 KB	Output is correct
85	Correct	5 ms	11988 KB	Output is correct
86	Correct	21 ms	14548 KB	Output is correct
87	Correct	21 ms	14580 KB	Output is correct
88	Correct	22 ms	14676 KB	Output is correct
89	Correct	10 ms	15028 KB	Output is correct
90	Correct	10 ms	14592 KB	Output is correct
91	Correct	22 ms	14648 KB	Output is correct
92	Correct	22 ms	14548 KB	Output is correct
93	Correct	6 ms	11988 KB	Output is correct
94	Correct	89 ms	45380 KB	Output is correct
95	Correct	94 ms	44900 KB	Output is correct
96	Correct	94 ms	44116 KB	Output is correct
97	Correct	89 ms	45376 KB	Output is correct
98	Correct	77 ms	45632 KB	Output is correct
99	Correct	76 ms	36788 KB	Output is correct
100	Correct	74 ms	36688 KB	Output is correct
101	Correct	76 ms	36752 KB	Output is correct
102	Correct	74 ms	37068 KB	Output is correct
103	Correct	72 ms	40776 KB	Output is correct
104	Correct	73 ms	41180 KB	Output is correct
105	Correct	72 ms	40604 KB	Output is correct
106	Correct	85 ms	41092 KB	Output is correct
107	Correct	87 ms	40896 KB	Output is correct
108	Correct	83 ms	45588 KB	Output is correct
109	Correct	78 ms	45552 KB	Output is correct
110	Correct	62 ms	36372 KB	Output is correct
111	Correct	62 ms	36448 KB	Output is correct
112	Correct	76 ms	36684 KB	Output is correct
113	Correct	76 ms	36772 KB	Output is correct

답안 #758791

Compilation message

Subtask #1 8.0 / 8.0

Subtask #2 15.0 / 15.0

Subtask #3 7.0 / 7.0

Subtask #4 29.0 / 29.0

Subtask #5 16.0 / 16.0

Subtask #6 25.0 / 25.0

Subtask #1
8.0 / 8.0

Subtask #2
15.0 / 15.0

Subtask #3
7.0 / 7.0

Subtask #4
29.0 / 29.0

Subtask #5
16.0 / 16.0

Subtask #6
25.0 / 25.0