oj.uz

답안 #1003850

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1003850 2024-06-20T19:01:03 Z vjudge1 Pastiri (COI20_pastiri) C++17
49 / 100
 1000 ms 72480 KB 
#include <bits/stdc++.h>
using namespace std;

#define fi first
#define se second
#define pb push_back
#define all(x) x.begin(), x.end()
#define allr(x) x.rbegin(), x.rend()
#define mp make_pair

typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef complex<double> cd;

const int MAXN = 5e5+10;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const double PI = acos(-1);

vector<int> adj[MAXN];
int prof[MAXN], dist[MAXN], elim[MAXN], pai[MAXN];

void build(int ver, int prev=0){
	prof[ver]=prof[prev]+1;
	pai[ver]=prev;
	
	for(auto u : adj[ver]){
		if(u==prev) continue;
		build(u, ver);
	}
}

void dfs(int ver, int val, int lim, int prev=-1){
	elim[ver]=1;
	if(val==lim) return;
	
	for(auto u : adj[ver]){
		if(u==prev) continue;
		dfs(u, val+1, lim, ver);
	}
}

void solve(){
	int n, k; cin >> n >> k;
	
	for(int i=0;i<n-1;i++){
		int a, b; cin >> a >> b;
		adj[a].pb(b);
		adj[b].pb(a);
	}
	
	for(int i=1;i<=n;i++)
		dist[i]=-1;
	
	vector<int> spec(k);
	queue<int> q;
	for(int i=0;i<k;i++){
		cin >> spec[i];
		q.push(spec[i]);
		dist[spec[i]]=0;
	}
	
	while(!q.empty()){
		int curr=q.front();
		q.pop();
		
		for(auto u : adj[curr]){
			if(dist[u]==-1){
				dist[u]=dist[curr]+1;
				q.push(u);
			}
		}
	}
	
	build(1);
	sort(all(spec), [&](int a, int b){ return prof[a]>prof[b]; });
	
	vector<int> ans;
	for(auto u : spec){
		if(elim[u]) continue;
		
		int at=u, d=0;
		while(pai[at]!=0 && dist[pai[at]]==d+1){
			d++;
			at=pai[at];
		}
		
		ans.pb(at);
		dfs(at, 0, d);
	}
	
	cout << (int)ans.size() << "\n";
	for(int i=0;i<(int)ans.size();i++) cout << ans[i] << " \n"[i==(int)ans.size()-1];
}

int32_t main(){
	ios_base::sync_with_stdio(false); cin.tie(NULL);
	
	int tt=1;
	//~ cin >> tt;
	while(tt--) solve();
	return 0;
}

Subtask #1
8.0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 93 ms 57564 KB Output is correct
2 Correct 121 ms 65528 KB Output is correct
3 Correct 131 ms 65620 KB Output is correct
4 Correct 167 ms 72480 KB Output is correct

Subtask #2
18.0 / 18.0

# 결과 실행 시간 메모리 Grader output
1 Correct 6 ms 12380 KB Output is correct
2 Correct 7 ms 12340 KB Output is correct
3 Correct 188 ms 42816 KB Output is correct
4 Correct 197 ms 45136 KB Output is correct
5 Correct 262 ms 42268 KB Output is correct
6 Correct 649 ms 44484 KB Output is correct

Subtask #3
23.0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5 ms 12124 KB Output is correct
2 Correct 5 ms 12124 KB Output is correct
3 Correct 5 ms 12148 KB Output is correct
4 Correct 5 ms 12124 KB Output is correct
5 Correct 5 ms 12184 KB Output is correct
6 Correct 5 ms 12124 KB Output is correct
7 Correct 5 ms 12124 KB Output is correct
8 Correct 5 ms 12124 KB Output is correct
9 Correct 5 ms 12124 KB Output is correct
10 Correct 5 ms 12124 KB Output is correct
11 Correct 5 ms 12120 KB Output is correct
12 Correct 4 ms 12300 KB Output is correct
13 Correct 5 ms 12124 KB Output is correct
14 Correct 6 ms 12380 KB Output is correct
15 Correct 5 ms 12124 KB Output is correct

Subtask #4
0 / 51.0

# 결과 실행 시간 메모리 Grader output
1 Correct 249 ms 36436 KB Output is correct
2 Correct 210 ms 43088 KB Output is correct
3 Correct 223 ms 46624 KB Output is correct
4 Correct 266 ms 42068 KB Output is correct
5 Correct 164 ms 40600 KB Output is correct
6 Correct 240 ms 50332 KB Output is correct
7 Correct 272 ms 48724 KB Output is correct
8 Correct 270 ms 48212 KB Output is correct
9 Execution timed out 1063 ms 42324 KB Time limit exceeded
10 Halted 0 ms 0 KB -