Submission #1325612

#TimeUsernameProblemLanguageResultExecution timeMemory
1325612boclobanchatMousetrap (CEOI17_mousetrap)C++20
100 / 100
352 ms149152 KiB
#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e6+67;
const int INF=1e9;
vector<int> ds[MAXN];
int dp[MAXN],root[MAXN],pref[MAXN],vv[MAXN];
bool del[MAXN];
void dfs(int i,int pre)
{
	root[i]=pre;
	dp[i]=ds[i].size()-(i!=pre);
	int mx1=0,mx2=0;
	for(auto v:ds[i]) if(v!=pre)
	{
		dfs(v,i);
		if(mx1<dp[v]) mx2=mx1,mx1=dp[v];
		else if(mx2<dp[v]) mx2=dp[v];
	}
	dp[i]+=mx2;
}
int fastinput()
{
	char res=' ';
	int ans=0;
	while(!(res>='0'&&res<='9')) res=getchar();
	while(res>='0'&&res<='9') ans=ans*10+res-'0',res=getchar();
	return ans;
}
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int n=fastinput(),t=fastinput(),m=fastinput();
	for(int i=1;i<n;i++)
	{
		int l=fastinput(),r=fastinput();
		ds[l].push_back(r),ds[r].push_back(l);
	}
	dfs(t,t);
	int pos=m,cc=0;
	bool c=false;
	while(pos!=t) vv[++cc]=pos,pref[pos]=ds[pos].size()-1-c,c=true,pos=root[pos];
	for(int i=cc-1;i;i--) pref[vv[i]]+=pref[vv[i+1]];
	int l=0,r=n,ans=n;
	while(l<=r)
	{
		int mid=(l+r)/2,res=m,cnt=1,z=0,pre=0;
		bool ck=true;
		while(res!=t)
		{
			int k=0;
			for(auto v:ds[res]) if(v!=pre&&v!=root[res]&&dp[v]+pref[res]+z>mid)
			{
				cnt--,k++;
				if(cnt<0)
				{
					ck=false;
					break;
				}
			}
			cnt++,z+=k,pre=res,res=root[res];
		}
		if(ck&&z<=mid) r=mid-1,ans=mid;
		else l=mid+1;
	}
	cout<<ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...