답안 #102380

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
102380	2019-03-24T14:50:09 Z	Dat160601	Mergers (JOI19_mergers)	C++17	70 / 100	3000 ms	208220 KB

mergers

#include <bits/stdc++.h>
using namespace std;
#define mp make_pair
#define pb push_back
#define fi first
#define se second

const int N = 5e5 + 7;

int n, k, u, v, s[N], level[N], sub[N], par[20][N], cnt = 0, pset[N], ret[N], sz[N], ans = 0;
bool vis[N], col[N];
vector <int> edge[N], tp[N], g[N], leaf;
map < pair <int, int>, int > ed;

void predfs(int u, int p){
	int ch = 0;
	for(int v : edge[u]){
		if(v == p) continue;
		par[0][v] = u;
		level[v] = level[u] + 1;
		predfs(v, u);
		ch++;
	}
	if(ch == 0) leaf.pb(u);
}

int lca(int u, int v){
	if(level[u] > level[v]) swap(u, v);
	for(int i = 19; i >= 0; i--) if(level[par[i][v]] >= level[u]) v = par[i][v];
	for(int i = 19; i >= 0; i--) if(par[i][u] != par[i][v]) u = par[i][u], v = par[i][v];
	if(u == v) return u;
	return par[0][u];
}

int fset(int x){
	if(pset[x] == x) return x;
	return pset[x] = fset(pset[x]);
}

void unionset(int u, int v){
	u = fset(u), v = fset(v);
	if(u == v) return;
	pset[u] = v;
	sz[v] += sz[u];
	sz[u] = 0;
}

void dfs(int u, int p){
	int child = 0;
	for(int v : g[u]){
		if(v == p) continue;
		child ++;
		dfs(v, u);
	}
	if(child == 0 || (child == 1 && u == 1)) ans++;
}

int main(){
	ios_base::sync_with_stdio(0);
	cin >> n >> k;
	for(int i = 1; i < n; i++){
		cin >> u >> v;
		edge[u].pb(v);
		edge[v].pb(u);
	}
	for(int i = 1; i <= n; i++){
		cin >> s[i];
		tp[s[i]].pb(i);
	}
	par[0][1] = 1;
	level[1] = 1;
	predfs(1, 1);
	for(int i = 1; i <= 19; i++){
		for(int j = 1; j <= n; j++){
			par[i][j] = par[i - 1][par[i - 1][j]];
		}
	}
	for(int i = 1; i <= k; i++){
		pset[i] = i;
		sz[i] = 1;
		if(tp[i].empty()) continue;
		int cur = tp[i][0];
		for(int j = 1; j < (int)tp[i].size(); j++){
			cur = lca(cur, tp[i][j]);
		}
		sub[cur]++;
	}
	priority_queue < pair <int, int> > q;
	for(int x : leaf) vis[x] = true, q.push(mp(level[x], x));
	while(!q.empty()){
		int u = q.top().se;
		q.pop();
		int p = fset(s[u]);
		sz[p] -= sub[u];
		if(!vis[par[0][u]]){
			vis[par[0][u]] = true;
			q.push(mp(level[par[0][u]], par[0][u]));
		}
		if(sz[fset(p)] == 0) continue;
		unionset(p, s[par[0][u]]);
	}
	for(int i = 1; i <= k; i++){
		if(pset[i] == i) ret[i] = ++cnt;
	}
	for(int i = 1; i <= k; i++){
		if(pset[i] != i) ret[i] = ret[fset(i)];
	}
	for(int i = 1; i <= n; i++){
		int id = ret[s[i]];
		for(int j : edge[i]){
			int jd = ret[s[j]];
			if(id == jd) continue;
			if(ed.count(mp(id, jd))) continue;
			g[id].pb(jd);
			g[jd].pb(id);
			ed[mp(id, jd)]++;
			ed[mp(jd, id)]++;
		}
	}
	dfs(1, 1);
	if(ans == 1) cout << 0;
	else cout << (ans + 1) / 2;
}

Subtask #1
10.0 / 10.0

#	결과	실행 시간	메모리	Grader output
1	Correct	34 ms	35840 KB	Output is correct
2	Correct	36 ms	35836 KB	Output is correct
3	Correct	38 ms	35828 KB	Output is correct
4	Correct	34 ms	35840 KB	Output is correct
5	Correct	33 ms	35960 KB	Output is correct
6	Correct	33 ms	35960 KB	Output is correct
7	Correct	35 ms	35704 KB	Output is correct
8	Correct	41 ms	35832 KB	Output is correct
9	Correct	43 ms	35732 KB	Output is correct
10	Correct	31 ms	35708 KB	Output is correct
11	Correct	39 ms	35840 KB	Output is correct
12	Correct	41 ms	35832 KB	Output is correct
13	Correct	33 ms	35704 KB	Output is correct
14	Correct	40 ms	35772 KB	Output is correct
15	Correct	34 ms	35832 KB	Output is correct
16	Correct	30 ms	35704 KB	Output is correct
17	Correct	32 ms	35704 KB	Output is correct
18	Correct	31 ms	35840 KB	Output is correct
19	Correct	37 ms	35824 KB	Output is correct
20	Correct	36 ms	35832 KB	Output is correct
21	Correct	35 ms	35840 KB	Output is correct
22	Correct	37 ms	35832 KB	Output is correct
23	Correct	37 ms	35708 KB	Output is correct
24	Correct	35 ms	35720 KB	Output is correct

Subtask #2
24.0 / 24.0

#	결과	실행 시간	메모리	Grader output
1	Correct	34 ms	35840 KB	Output is correct
2	Correct	36 ms	35836 KB	Output is correct
3	Correct	38 ms	35828 KB	Output is correct
4	Correct	34 ms	35840 KB	Output is correct
5	Correct	33 ms	35960 KB	Output is correct
6	Correct	33 ms	35960 KB	Output is correct
7	Correct	35 ms	35704 KB	Output is correct
8	Correct	41 ms	35832 KB	Output is correct
9	Correct	43 ms	35732 KB	Output is correct
10	Correct	31 ms	35708 KB	Output is correct
11	Correct	39 ms	35840 KB	Output is correct
12	Correct	41 ms	35832 KB	Output is correct
13	Correct	33 ms	35704 KB	Output is correct
14	Correct	40 ms	35772 KB	Output is correct
15	Correct	34 ms	35832 KB	Output is correct
16	Correct	30 ms	35704 KB	Output is correct
17	Correct	32 ms	35704 KB	Output is correct
18	Correct	31 ms	35840 KB	Output is correct
19	Correct	37 ms	35824 KB	Output is correct
20	Correct	36 ms	35832 KB	Output is correct
21	Correct	35 ms	35840 KB	Output is correct
22	Correct	37 ms	35832 KB	Output is correct
23	Correct	37 ms	35708 KB	Output is correct
24	Correct	35 ms	35720 KB	Output is correct
25	Correct	32 ms	35832 KB	Output is correct
26	Correct	38 ms	36472 KB	Output is correct
27	Correct	42 ms	36216 KB	Output is correct
28	Correct	41 ms	36700 KB	Output is correct
29	Correct	38 ms	36728 KB	Output is correct
30	Correct	37 ms	36224 KB	Output is correct
31	Correct	33 ms	35840 KB	Output is correct
32	Correct	39 ms	36480 KB	Output is correct
33	Correct	35 ms	35740 KB	Output is correct
34	Correct	35 ms	36216 KB	Output is correct
35	Correct	41 ms	36728 KB	Output is correct
36	Correct	38 ms	36216 KB	Output is correct
37	Correct	37 ms	36224 KB	Output is correct
38	Correct	36 ms	35832 KB	Output is correct
39	Correct	32 ms	36260 KB	Output is correct
40	Correct	38 ms	36236 KB	Output is correct
41	Correct	44 ms	36220 KB	Output is correct
42	Correct	38 ms	36348 KB	Output is correct
43	Correct	37 ms	36352 KB	Output is correct
44	Correct	38 ms	35832 KB	Output is correct
45	Correct	39 ms	36336 KB	Output is correct
46	Correct	36 ms	36348 KB	Output is correct
47	Correct	39 ms	35832 KB	Output is correct
48	Correct	35 ms	36216 KB	Output is correct
49	Correct	42 ms	36868 KB	Output is correct
50	Correct	44 ms	36856 KB	Output is correct
51	Correct	40 ms	36224 KB	Output is correct
52	Correct	40 ms	36216 KB	Output is correct
53	Correct	41 ms	36352 KB	Output is correct
54	Correct	38 ms	36268 KB	Output is correct

Subtask #3
14.0 / 14.0

#	결과	실행 시간	메모리	Grader output
1	Correct	34 ms	35840 KB	Output is correct
2	Correct	36 ms	35836 KB	Output is correct
3	Correct	38 ms	35828 KB	Output is correct
4	Correct	34 ms	35840 KB	Output is correct
5	Correct	33 ms	35960 KB	Output is correct
6	Correct	33 ms	35960 KB	Output is correct
7	Correct	35 ms	35704 KB	Output is correct
8	Correct	41 ms	35832 KB	Output is correct
9	Correct	43 ms	35732 KB	Output is correct
10	Correct	31 ms	35708 KB	Output is correct
11	Correct	39 ms	35840 KB	Output is correct
12	Correct	41 ms	35832 KB	Output is correct
13	Correct	33 ms	35704 KB	Output is correct
14	Correct	40 ms	35772 KB	Output is correct
15	Correct	34 ms	35832 KB	Output is correct
16	Correct	30 ms	35704 KB	Output is correct
17	Correct	32 ms	35704 KB	Output is correct
18	Correct	31 ms	35840 KB	Output is correct
19	Correct	37 ms	35824 KB	Output is correct
20	Correct	36 ms	35832 KB	Output is correct
21	Correct	35 ms	35840 KB	Output is correct
22	Correct	37 ms	35832 KB	Output is correct
23	Correct	37 ms	35708 KB	Output is correct
24	Correct	35 ms	35720 KB	Output is correct
25	Correct	36 ms	35804 KB	Output is correct
26	Correct	317 ms	50156 KB	Output is correct
27	Correct	360 ms	49088 KB	Output is correct
28	Correct	37 ms	36216 KB	Output is correct
29	Correct	34 ms	35876 KB	Output is correct
30	Correct	32 ms	35840 KB	Output is correct
31	Correct	284 ms	49180 KB	Output is correct
32	Correct	37 ms	36216 KB	Output is correct
33	Correct	183 ms	54028 KB	Output is correct
34	Correct	314 ms	49080 KB	Output is correct
35	Correct	37 ms	36216 KB	Output is correct
36	Correct	359 ms	48888 KB	Output is correct
37	Correct	32 ms	36224 KB	Output is correct
38	Correct	38 ms	36216 KB	Output is correct
39	Correct	337 ms	50064 KB	Output is correct
40	Correct	38 ms	36344 KB	Output is correct
41	Correct	180 ms	49236 KB	Output is correct
42	Correct	218 ms	50044 KB	Output is correct
43	Correct	32 ms	35840 KB	Output is correct
44	Correct	199 ms	54228 KB	Output is correct
45	Correct	191 ms	51972 KB	Output is correct
46	Correct	34 ms	36216 KB	Output is correct
47	Correct	35 ms	36216 KB	Output is correct

Subtask #4
22.0 / 22.0

#	결과	실행 시간	메모리	Grader output
1	Correct	264 ms	50044 KB	Output is correct
2	Correct	604 ms	69924 KB	Output is correct
3	Correct	40 ms	36480 KB	Output is correct
4	Correct	37 ms	36188 KB	Output is correct
5	Correct	37 ms	35832 KB	Output is correct
6	Correct	38 ms	35812 KB	Output is correct
7	Correct	36 ms	36216 KB	Output is correct
8	Correct	559 ms	52572 KB	Output is correct
9	Correct	34 ms	36216 KB	Output is correct
10	Correct	439 ms	49516 KB	Output is correct
11	Correct	38 ms	35832 KB	Output is correct
12	Correct	431 ms	48832 KB	Output is correct
13	Correct	498 ms	53892 KB	Output is correct
14	Correct	480 ms	68720 KB	Output is correct
15	Correct	265 ms	50032 KB	Output is correct
16	Correct	40 ms	36344 KB	Output is correct
17	Correct	36 ms	35832 KB	Output is correct
18	Correct	469 ms	66404 KB	Output is correct
19	Correct	509 ms	72308 KB	Output is correct
20	Correct	36 ms	36344 KB	Output is correct
21	Correct	34 ms	35832 KB	Output is correct
22	Correct	422 ms	54692 KB	Output is correct
23	Correct	35 ms	36216 KB	Output is correct
24	Correct	406 ms	50752 KB	Output is correct
25	Correct	582 ms	70768 KB	Output is correct
26	Correct	38 ms	36860 KB	Output is correct
27	Correct	45 ms	36980 KB	Output is correct
28	Correct	37 ms	36216 KB	Output is correct
29	Correct	37 ms	36224 KB	Output is correct
30	Correct	38 ms	36296 KB	Output is correct

Subtask #5
0 / 30.0

#	결과	실행 시간	메모리	Grader output
1	Correct	34 ms	35840 KB	Output is correct
2	Correct	36 ms	35836 KB	Output is correct
3	Correct	38 ms	35828 KB	Output is correct
4	Correct	34 ms	35840 KB	Output is correct
5	Correct	33 ms	35960 KB	Output is correct
6	Correct	33 ms	35960 KB	Output is correct
7	Correct	35 ms	35704 KB	Output is correct
8	Correct	41 ms	35832 KB	Output is correct
9	Correct	43 ms	35732 KB	Output is correct
10	Correct	31 ms	35708 KB	Output is correct
11	Correct	39 ms	35840 KB	Output is correct
12	Correct	41 ms	35832 KB	Output is correct
13	Correct	33 ms	35704 KB	Output is correct
14	Correct	40 ms	35772 KB	Output is correct
15	Correct	34 ms	35832 KB	Output is correct
16	Correct	30 ms	35704 KB	Output is correct
17	Correct	32 ms	35704 KB	Output is correct
18	Correct	31 ms	35840 KB	Output is correct
19	Correct	37 ms	35824 KB	Output is correct
20	Correct	36 ms	35832 KB	Output is correct
21	Correct	35 ms	35840 KB	Output is correct
22	Correct	37 ms	35832 KB	Output is correct
23	Correct	37 ms	35708 KB	Output is correct
24	Correct	35 ms	35720 KB	Output is correct
25	Correct	32 ms	35832 KB	Output is correct
26	Correct	38 ms	36472 KB	Output is correct
27	Correct	42 ms	36216 KB	Output is correct
28	Correct	41 ms	36700 KB	Output is correct
29	Correct	38 ms	36728 KB	Output is correct
30	Correct	37 ms	36224 KB	Output is correct
31	Correct	33 ms	35840 KB	Output is correct
32	Correct	39 ms	36480 KB	Output is correct
33	Correct	35 ms	35740 KB	Output is correct
34	Correct	35 ms	36216 KB	Output is correct
35	Correct	41 ms	36728 KB	Output is correct
36	Correct	38 ms	36216 KB	Output is correct
37	Correct	37 ms	36224 KB	Output is correct
38	Correct	36 ms	35832 KB	Output is correct
39	Correct	32 ms	36260 KB	Output is correct
40	Correct	38 ms	36236 KB	Output is correct
41	Correct	44 ms	36220 KB	Output is correct
42	Correct	38 ms	36348 KB	Output is correct
43	Correct	37 ms	36352 KB	Output is correct
44	Correct	38 ms	35832 KB	Output is correct
45	Correct	39 ms	36336 KB	Output is correct
46	Correct	36 ms	36348 KB	Output is correct
47	Correct	39 ms	35832 KB	Output is correct
48	Correct	35 ms	36216 KB	Output is correct
49	Correct	42 ms	36868 KB	Output is correct
50	Correct	44 ms	36856 KB	Output is correct
51	Correct	40 ms	36224 KB	Output is correct
52	Correct	40 ms	36216 KB	Output is correct
53	Correct	41 ms	36352 KB	Output is correct
54	Correct	38 ms	36268 KB	Output is correct
55	Correct	36 ms	35804 KB	Output is correct
56	Correct	317 ms	50156 KB	Output is correct
57	Correct	360 ms	49088 KB	Output is correct
58	Correct	37 ms	36216 KB	Output is correct
59	Correct	34 ms	35876 KB	Output is correct
60	Correct	32 ms	35840 KB	Output is correct
61	Correct	284 ms	49180 KB	Output is correct
62	Correct	37 ms	36216 KB	Output is correct
63	Correct	183 ms	54028 KB	Output is correct
64	Correct	314 ms	49080 KB	Output is correct
65	Correct	37 ms	36216 KB	Output is correct
66	Correct	359 ms	48888 KB	Output is correct
67	Correct	32 ms	36224 KB	Output is correct
68	Correct	38 ms	36216 KB	Output is correct
69	Correct	337 ms	50064 KB	Output is correct
70	Correct	38 ms	36344 KB	Output is correct
71	Correct	180 ms	49236 KB	Output is correct
72	Correct	218 ms	50044 KB	Output is correct
73	Correct	32 ms	35840 KB	Output is correct
74	Correct	199 ms	54228 KB	Output is correct
75	Correct	191 ms	51972 KB	Output is correct
76	Correct	34 ms	36216 KB	Output is correct
77	Correct	35 ms	36216 KB	Output is correct
78	Correct	264 ms	50044 KB	Output is correct
79	Correct	604 ms	69924 KB	Output is correct
80	Correct	40 ms	36480 KB	Output is correct
81	Correct	37 ms	36188 KB	Output is correct
82	Correct	37 ms	35832 KB	Output is correct
83	Correct	38 ms	35812 KB	Output is correct
84	Correct	36 ms	36216 KB	Output is correct
85	Correct	559 ms	52572 KB	Output is correct
86	Correct	34 ms	36216 KB	Output is correct
87	Correct	439 ms	49516 KB	Output is correct
88	Correct	38 ms	35832 KB	Output is correct
89	Correct	431 ms	48832 KB	Output is correct
90	Correct	498 ms	53892 KB	Output is correct
91	Correct	480 ms	68720 KB	Output is correct
92	Correct	265 ms	50032 KB	Output is correct
93	Correct	40 ms	36344 KB	Output is correct
94	Correct	36 ms	35832 KB	Output is correct
95	Correct	469 ms	66404 KB	Output is correct
96	Correct	509 ms	72308 KB	Output is correct
97	Correct	36 ms	36344 KB	Output is correct
98	Correct	34 ms	35832 KB	Output is correct
99	Correct	422 ms	54692 KB	Output is correct
100	Correct	35 ms	36216 KB	Output is correct
101	Correct	406 ms	50752 KB	Output is correct
102	Correct	582 ms	70768 KB	Output is correct
103	Correct	38 ms	36860 KB	Output is correct
104	Correct	45 ms	36980 KB	Output is correct
105	Correct	37 ms	36216 KB	Output is correct
106	Correct	37 ms	36224 KB	Output is correct
107	Correct	38 ms	36296 KB	Output is correct
108	Correct	2717 ms	120796 KB	Output is correct
109	Correct	1529 ms	111792 KB	Output is correct
110	Correct	1034 ms	120652 KB	Output is correct
111	Execution timed out	3041 ms	208220 KB	Time limit exceeded
112	Halted	0 ms	0 KB	-

답안 #102380

Compilation message

Subtask #1 10.0 / 10.0

Subtask #2 24.0 / 24.0

Subtask #3 14.0 / 14.0

Subtask #4 22.0 / 22.0

Subtask #5 0 / 30.0

Subtask #1
10.0 / 10.0

Subtask #2
24.0 / 24.0

Subtask #3
14.0 / 14.0

Subtask #4
22.0 / 22.0

Subtask #5
0 / 30.0