oj.uz

Submission #698693

# Submission time Handle Problem Language Result Execution time Memory
698693 2023-02-14T07:58:57 Z moonrabbit2 Pastiri (COI20_pastiri) C++17
100 / 100
 754 ms 133376 KB 
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,sse4.1,sse4.2,popcnt,abm,mmx,avx,avx2,fma")
#include <bits/stdc++.h>
using namespace std;
const int N=5e5+5,inf=1e8;
int n,k;
vector<int> G[N],g[N];
int par[N];
bool in[N];
int sub[N],up[N],sz[N],best[N][2],dp[N][2][2];
vector<int> ans;
void dfs1(int u){
	sub[u]=inf;
	if(in[u]){
		sub[u]=0;
		sz[u]=1;
	}
	for(int v: G[u]) if(v!=par[u]){
		par[v]=u;
		g[u].emplace_back(v);
		dfs1(v);
		sub[u]=min(sub[u],sub[v]+1);
		sz[u]+=sz[v];
	}
}
void dfs2(int u){
	up[u]=min(up[u],inf);
	if(in[u]) up[u]=0;
	if(!g[u].size()) return;
	best[u][0]=g[u][0];
	for(int v: g[u]) if(sub[v]<sub[best[u][0]]) best[u][0]=v;
	best[u][1]=-1;
	for(int v: g[u]) if(v!=best[u][0]&&(best[u][1]==-1||sub[v]<sub[best[u][1]])) best[u][1]=v;
	for(int v: g[u]){
		if(v!=best[u][0]) up[v]=min(up[u]+1,sub[best[u][0]]+2);
		else{
			up[v]=up[u]+1;
			if(best[u][1]!=-1) up[v]=min(up[v],sub[best[u][1]]+2);
		}
		dfs2(v);
	}
}
void dfs3(int u){
	dp[u][0][0]=dp[u][0][1]=dp[u][1][0]=dp[u][1][1]=inf;
	for(int v: g[u]) dfs3(v);
	if(in[u]){
		int s=0;
		for(int v: g[u]) s+=dp[v][1][0];
		dp[u][0][0]=s;
		dp[u][1][0]=s+1;
		for(int v: g[u]) dp[u][1][0]=min(dp[u][1][0],s+dp[v][1][1]-dp[v][1][0]);
	} else{
		if(sub[u]==inf){
			dp[u][1][0]=0;
			dp[u][1][1]=1;
		} else{
			if(sub[u]==up[u]){
				{
					int s=0;
					for(int v: g[u]) s+=dp[v][1][0];
					dp[u][1][0]=min(dp[u][1][0],s);
				}
				{
					int s=0;
					for(int v: g[u]){
						if(sub[u]==sub[v]+1) s+=dp[v][0][0];
						else s+=dp[v][1][0];
					}
					dp[u][1][1]=min(dp[u][1][1],s+1);
					for(int v: g[u]) if(sub[u]!=sub[v]+1) dp[u][1][1]=min(dp[u][1][1],s-dp[v][1][0]+dp[v][1][1]);
				}
			} else if(sub[u]>up[u]){
				{
					int s=0;
					for(int v: g[u]) s+=dp[v][1][0];
					dp[u][1][0]=min(dp[u][1][0],s);
					dp[u][1][1]=min(dp[u][1][1],s+1);
					for(int v: g[u]) dp[u][1][1]=min(dp[u][1][1],s-dp[v][1][0]+dp[v][1][1]);
				}
			} else{
				{
					int s=0;
					for(int v: g[u]) s+=dp[v][1][0];
					dp[u][1][0]=min(dp[u][1][0],s);
				}
				{
					int s=0;
					for(int v: g[u]){
						if(sub[u]==sub[v]+1) s+=dp[v][0][0];
						else s+=dp[v][1][0];
					}
					if(sub[u]+1<=up[u]-1) dp[u][0][0]=min(dp[u][0][0],s);
					dp[u][1][0]=min(dp[u][1][0],s+1);
					for(int v: g[u]) if(sub[u]!=sub[v]+1) dp[u][1][0]=min(dp[u][1][0],s-dp[v][1][0]+dp[v][1][1]);
				}
			}
		}
	}
	dp[u][1][0]=min(dp[u][1][0],dp[u][1][1]);
	dp[u][0][1]=min(dp[u][0][1],dp[u][1][1]);
	dp[u][0][0]=min({dp[u][0][0],dp[u][0][1],dp[u][1][0]});
	if(sz[u]==k){
		dp[u][1][1]=min(dp[u][1][1],dp[u][1][0]);
		dp[u][0][1]=min(dp[u][0][1],dp[u][0][0]);
	}
}
void track(int u,int x,int y){
	if(!dp[u][x][y]) return;
	if(in[u]){
		int s=0;
		for(int v: g[u]) s+=dp[v][1][0];
		if(x==0&&y==0&&dp[u][x][y]==s){
			for(int v: g[u]) track(v,1,0);
			return;
		}
		if(x==1&&y==0&&dp[u][x][y]==s+1){
			ans.emplace_back(u);
			for(int v: g[u]) track(v,1,0);
			return;
		}
		for(int v: g[u]){
			if(x==1&&y==0&&dp[u][x][y]==s+dp[v][1][1]-dp[v][1][0]){
				for(int w: g[u]) if(v!=w) track(w,1,0);
				track(v,1,1);
				return;
			}
		}
	} else{
		if(sub[u]==inf){
			if(x==1&&y==0&&dp[u][x][y]==0) return;
			if(x==1&&y==1&&dp[u][x][y]==1){
				ans.emplace_back(u);
				return;
			}
		} else{
			if(sub[u]==up[u]){
				{
					int s=0;
					for(int v: g[u]) s+=dp[v][1][0];
					if(x==1&&y==0&&dp[u][x][y]==s){
						for(int v: g[u]) track(v,1,0);
						return;
					}
				}
				{
					int s=0;
					for(int v: g[u]){
						if(sub[u]==sub[v]+1) s+=dp[v][0][0];
						else s+=dp[v][1][0];
					}
					if(x==1&&y==1&&dp[u][x][y]==s+1){
						ans.emplace_back(u);
						for(int v: g[u]){
							if(sub[u]==sub[v]+1) track(v,0,0);
							else track(v,1,0);
						}
						return;
					}
					for(int v: g[u]) if(sub[u]!=sub[v]+1){
						if(x==1&&y==1&&dp[u][x][y]==s-dp[v][1][0]+dp[v][1][1]){
							for(int w: g[u]) if(v!=w){
								if(sub[u]==sub[w]+1) track(w,0,0);
								else track(w,1,0);
							}
							track(v,1,1);
							return;
						}
					}
				}
			} else if(sub[u]>up[u]){
				{
					int s=0;
					for(int v: g[u]) s+=dp[v][1][0];
					if(x==1&&y==0&&dp[u][x][y]==s){
						for(int v: g[u]) track(v,1,0);
						return;
					}
					if(x==1&&y==1&&dp[u][x][y]==s+1){
						ans.emplace_back(u);
						for(int v: g[u]) track(v,1,0);
						return;
					}
					for(int v: g[u]){
						if(x==1&&y==1&&dp[u][x][y]==s-dp[v][1][0]+dp[v][1][1]){
							for(int w: g[u]) if(v!=w) track(w,1,0);
							track(v,1,1);
							return;
						}
					}
				}
			} else{
				{
					int s=0;
					for(int v: g[u]) s+=dp[v][1][0];
					if(x==1&&y==0&&dp[u][x][y]==s){
						for(int v: g[u]) track(v,1,0);
						return;
					}
				}
				{
					int s=0;
					for(int v: g[u]){
						if(sub[u]==sub[v]+1) s+=dp[v][0][0];
						else s+=dp[v][1][0];
					}
					if(sub[u]+1<=up[u]-1){
						if(x==0&&y==0&&dp[u][x][y]==s){
							for(int v: g[u]){
								if(sub[u]==sub[v]+1) track(v,0,0);
								else track(v,1,0);
							}
							return;
						}
					}
					if(x==1&&y==0&&dp[u][x][y]==s+1){
						ans.emplace_back(u);
						for(int v: g[u]){
							if(sub[u]==sub[v]+1) track(v,0,0);
							else track(v,1,0);
						}
						return;
					}
					for(int v: g[u]) if(sub[u]!=sub[v]+1){
						if(x==1&&y==0&&dp[u][x][y]==s-dp[v][1][0]+dp[v][1][1]){
							for(int w: g[u]) if(v!=w){
								if(sub[u]==sub[w]+1) track(w,0,0);
								else track(w,1,0);
							}
							track(v,1,1);
							return;
						}
					}
				}
			}
		}
	}
	if(x==1&&y==0&&dp[u][x][y]==dp[u][1][1]){
		track(u,1,1);
		return;
	}
	if(x==0&&y==1&&dp[u][x][y]==dp[u][1][1]){
		track(u,1,1);
		return;
	}
	if(x==0&&y==0&&dp[u][x][y]==dp[u][0][1]){
		track(u,0,1);
		return;
	}
	if(x==0&&y==0&&dp[u][x][y]==dp[u][1][0]){
		track(u,1,0);
		return;
	}
	if(sz[u]==k){
		if(x==1&&y==1&&dp[u][x][y]==dp[u][1][0]){
			track(u,1,0);
			return;
		}
		if(x==0&&y==1&&dp[u][x][y]==dp[u][0][0]){
			track(u,0,0);
			return;
		}
	}
	assert(0);
}
int main(){
	ios::sync_with_stdio(false); cin.tie(0);
	cin>>n>>k;
	for(int u,v,i=1;i<n;i++){
		cin>>u>>v;
		G[u].emplace_back(v);
		G[v].emplace_back(u);
	}
	for(int u,i=1;i<=k;i++){
		cin>>u;
		in[u]=true;
	}
	dfs1(1);
	up[1]=inf;
	dfs2(1);
	dfs3(1);
	cout<<dp[1][1][1]<<"\n";
	track(1,1,1);
	for(int u: ans) cout<<u<<" ";
	return 0;
}

Subtask #1
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 226 ms 128232 KB Output is correct
2 Correct 254 ms 128724 KB Output is correct
3 Correct 245 ms 128772 KB Output is correct
4 Correct 276 ms 133376 KB Output is correct

Subtask #2
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 15 ms 24276 KB Output is correct
2 Correct 17 ms 24276 KB Output is correct
3 Correct 501 ms 70928 KB Output is correct
4 Correct 477 ms 71688 KB Output is correct
5 Correct 652 ms 81432 KB Output is correct
6 Correct 735 ms 85636 KB Output is correct

Subtask #3
23.0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 24020 KB Output is correct
2 Correct 17 ms 24008 KB Output is correct
3 Correct 15 ms 24020 KB Output is correct
4 Correct 14 ms 24020 KB Output is correct
5 Correct 15 ms 23980 KB Output is correct
6 Correct 14 ms 24020 KB Output is correct
7 Correct 14 ms 24080 KB Output is correct
8 Correct 15 ms 23956 KB Output is correct
9 Correct 16 ms 24020 KB Output is correct
10 Correct 18 ms 23980 KB Output is correct
11 Correct 15 ms 23956 KB Output is correct
12 Correct 14 ms 23948 KB Output is correct
13 Correct 15 ms 24008 KB Output is correct
14 Correct 15 ms 24084 KB Output is correct
15 Correct 16 ms 24080 KB Output is correct

Subtask #4
51.0 / 51.0

# Verdict Execution time Memory Grader output
1 Correct 646 ms 77216 KB Output is correct
2 Correct 618 ms 77136 KB Output is correct
3 Correct 695 ms 78200 KB Output is correct
4 Correct 721 ms 81764 KB Output is correct
5 Correct 522 ms 70272 KB Output is correct
6 Correct 580 ms 84256 KB Output is correct
7 Correct 628 ms 82716 KB Output is correct
8 Correct 645 ms 83736 KB Output is correct
9 Correct 754 ms 81888 KB Output is correct
10 Correct 665 ms 81548 KB Output is correct
11 Correct 438 ms 71484 KB Output is correct
12 Correct 366 ms 72656 KB Output is correct
13 Correct 409 ms 74212 KB Output is correct
14 Correct 345 ms 71564 KB Output is correct
15 Correct 77 ms 32752 KB Output is correct
16 Correct 660 ms 76852 KB Output is correct
17 Correct 437 ms 68348 KB Output is correct
18 Correct 567 ms 71304 KB Output is correct
19 Correct 696 ms 84036 KB Output is correct
20 Correct 749 ms 90608 KB Output is correct
21 Correct 582 ms 76132 KB Output is correct
22 Correct 596 ms 77268 KB Output is correct