답안 #403304

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
403304	2021-05-13T02:51:09 Z	oolimry	족보 (KOI18_family)	C++17	100 / 100	555 ms	59764 KB

family

#include <bits/stdc++.h>
#define all(x) (x).begin(), (x).end()
using namespace std;
typedef pair<int,int> ii;

int n1, n2, K; 
vector<int> child[2][300005];
int num[2][300005];
int rep[2][300005];
struct check{ int num, u, t; };
vector<check> stuff;

void dfs(int t, int u){
	if(u != 0 && u <= K){
		num[t][u] = 1;
		rep[t][u] = u;
	}
	else{
		for(int v : child[t][u]){
			dfs(t, v);
			num[t][u] += num[t][v];
			rep[t][u] = rep[t][v];
		}
		if(num[t][u] > 1) stuff.push_back({num[t][u], u, t});
		///idk why child[t][u].size() > 1 is needed .-.
	}
}

int p[300005];
int sz[300005];
int findSet(int u){
	if(p[u] == u) return u;
	else return p[u] = findSet(p[u]);
}
void unionSet(int u, int v){
	u = findSet(u), v = findSet(v);
	if(u == v) return;
	if(u&1) swap(u,v);
	p[u] = v;
	sz[v] += sz[u];
}
int getSz(int u){
	return sz[findSet(u)];
}

int main(){
	//freopen("i.txt","r",stdin);
	cin >> n1 >> n2 >> K;
	
	for(int i = 1;i <= n1;i++){
		int a; cin >> a;
		child[0][a].push_back(i);
	}
	for(int i = 1;i <= n2;i++){
		int a; cin >> a;
		child[1][a].push_back(i);
	}

	dfs(0, 0); dfs(1, 0);
	
	for(int i = 1;i <= K;i++){
		p[i] = i;
		sz[i] = 1;
	}
		
	sort(all(stuff), [&](check a, check b){ return a.num < b.num; });
	
	for(check C : stuff){
		
		for(int v : child[C.t][C.u]){
			unionSet(rep[C.t][C.u], rep[C.t][v]);
			if(((int) child[C.t][C.u].size() == 1)) assert(rep[C.t][C.u] == rep[C.t][v]);
		} 
		if((int) child[C.t][C.u].size() <= 1) continue;
		if(getSz(rep[C.t][C.u]) != C.num){
			cout << "NO";
			return 0;
		}
	}
	
	
	cout << "YES";
}

Subtask #1
17.0 / 17.0

#	결과	실행 시간	메모리	Grader output
1	Correct	9 ms	14412 KB	Output is correct
2	Correct	9 ms	14312 KB	Output is correct
3	Correct	8 ms	14428 KB	Output is correct
4	Correct	8 ms	14412 KB	Output is correct
5	Correct	8 ms	14412 KB	Output is correct
6	Correct	9 ms	14368 KB	Output is correct
7	Correct	8 ms	14412 KB	Output is correct
8	Correct	8 ms	14412 KB	Output is correct
9	Correct	9 ms	14428 KB	Output is correct
10	Correct	9 ms	14412 KB	Output is correct
11	Correct	9 ms	14412 KB	Output is correct
12	Correct	8 ms	14412 KB	Output is correct
13	Correct	9 ms	14412 KB	Output is correct
14	Correct	8 ms	14412 KB	Output is correct
15	Correct	9 ms	14412 KB	Output is correct
16	Correct	9 ms	14412 KB	Output is correct
17	Correct	9 ms	14412 KB	Output is correct
18	Correct	9 ms	14412 KB	Output is correct
19	Correct	8 ms	14412 KB	Output is correct
20	Correct	9 ms	14412 KB	Output is correct
21	Correct	8 ms	14412 KB	Output is correct
22	Correct	10 ms	14412 KB	Output is correct
23	Correct	9 ms	14424 KB	Output is correct
24	Correct	9 ms	14412 KB	Output is correct
25	Correct	9 ms	14412 KB	Output is correct
26	Correct	10 ms	14412 KB	Output is correct
27	Correct	9 ms	14412 KB	Output is correct
28	Correct	9 ms	14412 KB	Output is correct
29	Correct	8 ms	14412 KB	Output is correct
30	Correct	9 ms	14412 KB	Output is correct

Subtask #2
27.0 / 27.0

#	결과	실행 시간	메모리	Grader output
1	Correct	9 ms	14412 KB	Output is correct
2	Correct	9 ms	14312 KB	Output is correct
3	Correct	8 ms	14428 KB	Output is correct
4	Correct	8 ms	14412 KB	Output is correct
5	Correct	8 ms	14412 KB	Output is correct
6	Correct	9 ms	14368 KB	Output is correct
7	Correct	8 ms	14412 KB	Output is correct
8	Correct	8 ms	14412 KB	Output is correct
9	Correct	9 ms	14428 KB	Output is correct
10	Correct	9 ms	14412 KB	Output is correct
11	Correct	9 ms	14412 KB	Output is correct
12	Correct	8 ms	14412 KB	Output is correct
13	Correct	9 ms	14412 KB	Output is correct
14	Correct	8 ms	14412 KB	Output is correct
15	Correct	9 ms	14412 KB	Output is correct
16	Correct	9 ms	14412 KB	Output is correct
17	Correct	9 ms	14412 KB	Output is correct
18	Correct	9 ms	14412 KB	Output is correct
19	Correct	8 ms	14412 KB	Output is correct
20	Correct	9 ms	14412 KB	Output is correct
21	Correct	8 ms	14412 KB	Output is correct
22	Correct	10 ms	14412 KB	Output is correct
23	Correct	9 ms	14424 KB	Output is correct
24	Correct	9 ms	14412 KB	Output is correct
25	Correct	9 ms	14412 KB	Output is correct
26	Correct	10 ms	14412 KB	Output is correct
27	Correct	9 ms	14412 KB	Output is correct
28	Correct	9 ms	14412 KB	Output is correct
29	Correct	8 ms	14412 KB	Output is correct
30	Correct	9 ms	14412 KB	Output is correct
31	Correct	8 ms	14412 KB	Output is correct
32	Correct	8 ms	14412 KB	Output is correct
33	Correct	9 ms	14412 KB	Output is correct
34	Correct	9 ms	14540 KB	Output is correct
35	Correct	9 ms	14364 KB	Output is correct
36	Correct	9 ms	14412 KB	Output is correct
37	Correct	9 ms	14360 KB	Output is correct
38	Correct	8 ms	14412 KB	Output is correct
39	Correct	9 ms	14412 KB	Output is correct
40	Correct	9 ms	14412 KB	Output is correct
41	Correct	9 ms	14412 KB	Output is correct
42	Correct	8 ms	14412 KB	Output is correct
43	Correct	8 ms	14420 KB	Output is correct
44	Correct	9 ms	14412 KB	Output is correct
45	Correct	9 ms	14412 KB	Output is correct
46	Correct	9 ms	14380 KB	Output is correct
47	Correct	9 ms	14412 KB	Output is correct
48	Correct	9 ms	14412 KB	Output is correct
49	Correct	9 ms	14412 KB	Output is correct
50	Correct	10 ms	14412 KB	Output is correct

Subtask #3
15.0 / 15.0

#	결과	실행 시간	메모리	Grader output
1	Correct	9 ms	14412 KB	Output is correct
2	Correct	9 ms	14312 KB	Output is correct
3	Correct	8 ms	14428 KB	Output is correct
4	Correct	8 ms	14412 KB	Output is correct
5	Correct	8 ms	14412 KB	Output is correct
6	Correct	9 ms	14368 KB	Output is correct
7	Correct	8 ms	14412 KB	Output is correct
8	Correct	8 ms	14412 KB	Output is correct
9	Correct	9 ms	14428 KB	Output is correct
10	Correct	9 ms	14412 KB	Output is correct
11	Correct	9 ms	14412 KB	Output is correct
12	Correct	8 ms	14412 KB	Output is correct
13	Correct	9 ms	14412 KB	Output is correct
14	Correct	8 ms	14412 KB	Output is correct
15	Correct	9 ms	14412 KB	Output is correct
16	Correct	9 ms	14412 KB	Output is correct
17	Correct	9 ms	14412 KB	Output is correct
18	Correct	9 ms	14412 KB	Output is correct
19	Correct	8 ms	14412 KB	Output is correct
20	Correct	9 ms	14412 KB	Output is correct
21	Correct	8 ms	14412 KB	Output is correct
22	Correct	10 ms	14412 KB	Output is correct
23	Correct	9 ms	14424 KB	Output is correct
24	Correct	9 ms	14412 KB	Output is correct
25	Correct	9 ms	14412 KB	Output is correct
26	Correct	10 ms	14412 KB	Output is correct
27	Correct	9 ms	14412 KB	Output is correct
28	Correct	9 ms	14412 KB	Output is correct
29	Correct	8 ms	14412 KB	Output is correct
30	Correct	9 ms	14412 KB	Output is correct
31	Correct	8 ms	14412 KB	Output is correct
32	Correct	8 ms	14412 KB	Output is correct
33	Correct	9 ms	14412 KB	Output is correct
34	Correct	9 ms	14540 KB	Output is correct
35	Correct	9 ms	14364 KB	Output is correct
36	Correct	9 ms	14412 KB	Output is correct
37	Correct	9 ms	14360 KB	Output is correct
38	Correct	8 ms	14412 KB	Output is correct
39	Correct	9 ms	14412 KB	Output is correct
40	Correct	9 ms	14412 KB	Output is correct
41	Correct	9 ms	14412 KB	Output is correct
42	Correct	8 ms	14412 KB	Output is correct
43	Correct	8 ms	14420 KB	Output is correct
44	Correct	9 ms	14412 KB	Output is correct
45	Correct	9 ms	14412 KB	Output is correct
46	Correct	9 ms	14380 KB	Output is correct
47	Correct	9 ms	14412 KB	Output is correct
48	Correct	9 ms	14412 KB	Output is correct
49	Correct	9 ms	14412 KB	Output is correct
50	Correct	10 ms	14412 KB	Output is correct
51	Correct	12 ms	14540 KB	Output is correct
52	Correct	13 ms	14608 KB	Output is correct
53	Correct	12 ms	14684 KB	Output is correct
54	Correct	12 ms	14584 KB	Output is correct
55	Correct	12 ms	14612 KB	Output is correct
56	Correct	13 ms	14528 KB	Output is correct
57	Correct	12 ms	14528 KB	Output is correct
58	Correct	12 ms	14540 KB	Output is correct
59	Correct	12 ms	14540 KB	Output is correct
60	Correct	11 ms	14540 KB	Output is correct
61	Correct	13 ms	15028 KB	Output is correct
62	Correct	13 ms	15132 KB	Output is correct
63	Correct	12 ms	14540 KB	Output is correct
64	Correct	11 ms	14532 KB	Output is correct
65	Correct	9 ms	14412 KB	Output is correct
66	Correct	13 ms	14708 KB	Output is correct
67	Correct	12 ms	14540 KB	Output is correct
68	Correct	11 ms	14696 KB	Output is correct
69	Correct	13 ms	14668 KB	Output is correct
70	Correct	12 ms	14540 KB	Output is correct

Subtask #4
41.0 / 41.0

#	결과	실행 시간	메모리	Grader output
1	Correct	9 ms	14412 KB	Output is correct
2	Correct	9 ms	14312 KB	Output is correct
3	Correct	8 ms	14428 KB	Output is correct
4	Correct	8 ms	14412 KB	Output is correct
5	Correct	8 ms	14412 KB	Output is correct
6	Correct	9 ms	14368 KB	Output is correct
7	Correct	8 ms	14412 KB	Output is correct
8	Correct	8 ms	14412 KB	Output is correct
9	Correct	9 ms	14428 KB	Output is correct
10	Correct	9 ms	14412 KB	Output is correct
11	Correct	9 ms	14412 KB	Output is correct
12	Correct	8 ms	14412 KB	Output is correct
13	Correct	9 ms	14412 KB	Output is correct
14	Correct	8 ms	14412 KB	Output is correct
15	Correct	9 ms	14412 KB	Output is correct
16	Correct	9 ms	14412 KB	Output is correct
17	Correct	9 ms	14412 KB	Output is correct
18	Correct	9 ms	14412 KB	Output is correct
19	Correct	8 ms	14412 KB	Output is correct
20	Correct	9 ms	14412 KB	Output is correct
21	Correct	8 ms	14412 KB	Output is correct
22	Correct	10 ms	14412 KB	Output is correct
23	Correct	9 ms	14424 KB	Output is correct
24	Correct	9 ms	14412 KB	Output is correct
25	Correct	9 ms	14412 KB	Output is correct
26	Correct	10 ms	14412 KB	Output is correct
27	Correct	9 ms	14412 KB	Output is correct
28	Correct	9 ms	14412 KB	Output is correct
29	Correct	8 ms	14412 KB	Output is correct
30	Correct	9 ms	14412 KB	Output is correct
31	Correct	8 ms	14412 KB	Output is correct
32	Correct	8 ms	14412 KB	Output is correct
33	Correct	9 ms	14412 KB	Output is correct
34	Correct	9 ms	14540 KB	Output is correct
35	Correct	9 ms	14364 KB	Output is correct
36	Correct	9 ms	14412 KB	Output is correct
37	Correct	9 ms	14360 KB	Output is correct
38	Correct	8 ms	14412 KB	Output is correct
39	Correct	9 ms	14412 KB	Output is correct
40	Correct	9 ms	14412 KB	Output is correct
41	Correct	9 ms	14412 KB	Output is correct
42	Correct	8 ms	14412 KB	Output is correct
43	Correct	8 ms	14420 KB	Output is correct
44	Correct	9 ms	14412 KB	Output is correct
45	Correct	9 ms	14412 KB	Output is correct
46	Correct	9 ms	14380 KB	Output is correct
47	Correct	9 ms	14412 KB	Output is correct
48	Correct	9 ms	14412 KB	Output is correct
49	Correct	9 ms	14412 KB	Output is correct
50	Correct	10 ms	14412 KB	Output is correct
51	Correct	12 ms	14540 KB	Output is correct
52	Correct	13 ms	14608 KB	Output is correct
53	Correct	12 ms	14684 KB	Output is correct
54	Correct	12 ms	14584 KB	Output is correct
55	Correct	12 ms	14612 KB	Output is correct
56	Correct	13 ms	14528 KB	Output is correct
57	Correct	12 ms	14528 KB	Output is correct
58	Correct	12 ms	14540 KB	Output is correct
59	Correct	12 ms	14540 KB	Output is correct
60	Correct	11 ms	14540 KB	Output is correct
61	Correct	13 ms	15028 KB	Output is correct
62	Correct	13 ms	15132 KB	Output is correct
63	Correct	12 ms	14540 KB	Output is correct
64	Correct	11 ms	14532 KB	Output is correct
65	Correct	9 ms	14412 KB	Output is correct
66	Correct	13 ms	14708 KB	Output is correct
67	Correct	12 ms	14540 KB	Output is correct
68	Correct	11 ms	14696 KB	Output is correct
69	Correct	13 ms	14668 KB	Output is correct
70	Correct	12 ms	14540 KB	Output is correct
71	Correct	59 ms	18272 KB	Output is correct
72	Correct	72 ms	17320 KB	Output is correct
73	Correct	38 ms	17156 KB	Output is correct
74	Correct	64 ms	21268 KB	Output is correct
75	Correct	87 ms	20672 KB	Output is correct
76	Correct	479 ms	59764 KB	Output is correct
77	Correct	555 ms	45788 KB	Output is correct
78	Correct	355 ms	27552 KB	Output is correct
79	Correct	449 ms	51196 KB	Output is correct
80	Correct	493 ms	44036 KB	Output is correct
81	Correct	53 ms	18228 KB	Output is correct
82	Correct	54 ms	17400 KB	Output is correct
83	Correct	37 ms	17296 KB	Output is correct
84	Correct	67 ms	20844 KB	Output is correct
85	Correct	61 ms	20672 KB	Output is correct
86	Correct	410 ms	53488 KB	Output is correct
87	Correct	427 ms	53568 KB	Output is correct
88	Correct	377 ms	27556 KB	Output is correct
89	Correct	425 ms	49700 KB	Output is correct
90	Correct	468 ms	44316 KB	Output is correct

답안 #403304

Compilation message

Subtask #1 17.0 / 17.0

Subtask #2 27.0 / 27.0

Subtask #3 15.0 / 15.0

Subtask #4 41.0 / 41.0

Subtask #1
17.0 / 17.0

Subtask #2
27.0 / 27.0

Subtask #3
15.0 / 15.0

Subtask #4
41.0 / 41.0