oj.uz

Submission #735585

# Submission time Handle Problem Language Result Execution time Memory
735585 2023-05-04T11:19:21 Z Koful123 Papričice (COCI20_papricice) C++17
0 / 110
 2 ms 3412 KB 
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl "\n"
#define ff first
#define ss second
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()

const int N = 1e5 + 5;
int ans = 1e18,n;
vector<int> adj[N],sz(N);
multiset<int> other,anc;

void calc(int node,int p){
	sz[node] = 1;
	for(int go : adj[node]){
		if(go == p) continue;
		calc(go,node);
		sz[node] = sz[go] + sz[node];
	}
}

int search(multiset<int> &s,int x,int ty){
	int res = 1e18,cur = x;
	if(!ty) cur = (n - x) / 2 + x;
	else cur = (n - x) / x;
	auto it = s.upper_bound(cur);
	if(ty){
		if(it != s.end()){
			vector<int> tmp = {x,n - x - *it,*it};
			sort(tmp.begin(), tmp.end());
			res = min(res,tmp[2] - tmp[0]);
		}
		if(it != s.begin()){
			vector<int> tmp = {x,n - *prev(it) - x,*prev(it)};
			sort(tmp.begin(), tmp.end());
			res = min(res,tmp[2] - tmp[0]);
		}
		return res;		
	}

	if(it != s.end()){
		vector<int> tmp = {x,n - *it,*it - x};
		sort(tmp.begin(), tmp.end());
		res = min(res,tmp[2] - tmp[0]);
	}
	if(it != s.begin()){
		vector<int> tmp = {x,n - *prev(it),*prev(it) - x};
		sort(tmp.begin(), tmp.end());
		res = min(res,tmp[2] - tmp[0]);
	}
	return res;
}

void dfs(int node,int p){
	ans = min({ans,search(anc,sz[node],0),search(other,sz[node],1)});
	anc.insert(sz[node]);
	for(int go : adj[node]){
		if(go == p) continue;
		dfs(go,node);
	}
	anc.erase(anc.find(sz[node]));
	other.insert(sz[node]);
}	

void solve(){

	cin >> n;
	for(int i = 1; i < n; i++){
		int a,b; cin >> a >> b;
		adj[a].push_back(b);
		adj[b].push_back(a);
	}

	calc(1,0);
	dfs(1,0);

	cout << ans << endl;
}
 
signed main(){ 
 
	ios::sync_with_stdio(0);
	cin.tie(0);
 
	int t = 1;
//	cin >> t;
 
	while(t--)
		solve();
 	
	return 0;
}

Subtask #1
0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 3412 KB Output is correct
2 Correct 2 ms 3412 KB Output is correct
3 Incorrect 2 ms 3412 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #2
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 3412 KB Output is correct
2 Correct 2 ms 3412 KB Output is correct
3 Incorrect 2 ms 3412 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 60.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 3412 KB Output is correct
2 Correct 2 ms 3412 KB Output is correct
3 Incorrect 2 ms 3412 KB Output isn't correct
4 Halted 0 ms 0 KB -