Submission #532874

#TimeUsernameProblemLanguageResultExecution timeMemory
532874Koosha_mvMeetings 2 (JOI21_meetings2)C++14
100 / 100
589 ms25808 KiB
#include <bits/stdc++.h>
using namespace std;
#define dbgv(v) cout<<#v<<" = "; f(i,0,v.size()) cout<<v[i]<<" "; cout<<endl
#define dbga(a,x,y) cout<<#a<<" = "; f(i,x,y) cout<<a[i]<<" "; cout<<endl
#define erorp(x) cout<<#x<<"={"<<(x.F)<<" , "<<x.S<<"}"<<endl
#define eror(x) cout<<#x<<'='<<(x)<<endl
#define f_(i,a,b) for(int i=a;i>=b;i--)
#define f(i,a,b) for(int i=a;i<b;i++)
#define nb(x) __builtin_popcount(x)
#define all(v) v.begin(),v.end()
#define bit(n,k) (((n)>>(k))&1)
#define Add(x,y) x=(x+y)%mod
#define maxm(a,b) a=max(a,b)
#define minm(a,b) a=min(a,b)
#define lst(x) x[x.size()-1]
#define sz(x) int(x.size())
#define mp make_pair
#define ll long long
#define pb push_back
#define S second
#define F first

const int N=2e5+99;

int n,t,prt,a[N],b[N],sz[N],mark[N],ans[N];
vector<int> g[N];

void dfs1(int u,int p,int s,int &cent){
	sz[u]=1;
	for(auto v : g[u]){
		if(v==p || mark[v]) continue ;
		dfs1(v,u,s,cent);
		sz[u]+=sz[v];
	}
	if(sz[u]>=s && (cent==0 || sz[u]<sz[cent])){
		cent=u;
	}
}
void dfs2(int u,int p,int h){
	maxm(a[sz[u]],h);
	for(auto v : g[u]){
		if(v==p || mark[v]) continue ;
		dfs2(v,u,h+1);
	}
}
void solve(int rt,int m=n){
	if(m==1) return ;
	int cent=0;
	dfs1(rt,rt,(m+1)/2,cent);
	dfs1(cent,cent,0,prt);
	mark[cent]=1;
	f(i,1,sz[cent]+1) b[i]=-N;
	for(auto v : g[cent]){
		if(mark[v]) continue ;
		f(i,1,sz[v]+1) a[i]=-N;
		dfs2(v,v,1);
		f_(i,sz[v]-1,1) maxm(a[i],a[i+1]);
		f(i,1,sz[v]+1){
			maxm(ans[i],a[i]+b[i]+1);
			maxm(b[i],a[i]);
			if(i<=m-sz[v]){
				maxm(ans[i],b[i]+1);
			}
		}
	}
	for(auto v : g[cent]){
		if(mark[v]) continue ;
		solve(v,sz[v]);
	}
}
main(){
	ios:: sync_with_stdio(0), cin.tie(0), cout.tie(0);
	cin>>n;
	f(i,1,n){
		int u,v;
		cin>>u>>v;
		g[u].pb(v);
		g[v].pb(u);
	}	
	solve(1);
	f(i,1,n+1){
		if(i&1) cout<<1<<'\n';
		else{
			cout<<max(1,ans[i/2])<<'\n';
		}
	}
}

Compilation message (stderr)

meetings2.cpp:71:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   71 | main(){
      | ^~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...