P3629 [APIO2010]巡逻 题解

const int N=1e5+13;
struct Edge{int v,nxt;}e[N<<1];
int n,k,h[N],etot=1;
inline void add_edge(int u,int v){e[++etot]=(Edge){v,h[u]};h[u]=etot;}

namespace sub1{
int ans,dep[N];
void dfs(int u,int fa){
	dep[u]=dep[fa]+1;
	for(int i=h[u];i;i=e[i].nxt){
		int v=e[i].v;if(v==fa) continue;
		dfs(v,u);
	}
}
inline void solve(){
	dfs(1,0);
	int rt=1;
	for(int i=2;i<=n;++i)
		if(dep[i]>dep[rt]) rt=i;
	dfs(rt,0);
	for(int i=1;i<=n;++i)
		if(dep[i]-1>ans) ans=dep[i]-1;
	println(2*(n-1)-ans+1);
}
}

namespace sub2{
int ans,dep[N],mx1[N],mx2[N];
bool vis[N<<1];
void dfs(int u,int fa){
	dep[u]=dep[fa]+1;
	for(int i=h[u];i;i=e[i].nxt){
		int v=e[i].v;if(v==fa) continue;
		dfs(v,u);
	}
}
bool dfs1(int u,int fa,int ed){
	if(u==ed) return 1;
	for(int i=h[u];i;i=e[i].nxt){
		int v=e[i].v;if(v==fa) continue;
		if(dfs1(v,u,ed)) return vis[i]=vis[i^1]=1;
	}
	return 0;
}
void dfs2(int u,int fa){
	mx1[u]=0,mx2[u]=-INF;
	for(int i=h[u];i;i=e[i].nxt){
		int v=e[i].v;if(v==fa) continue;
		int w=(vis[i]?-1:1);
		dfs2(v,u);
		if(mx1[v]+w>mx1[u]) mx2[u]=mx1[u],mx1[u]=mx1[v]+w;
		else if(mx1[v]+w>mx2[u]) mx2[u]=mx1[v]+w;
	}
	ans=max(ans,mx1[u]+mx2[u]);
}
inline void solve(){
	dfs(1,0);
	int x=1;
	for(int i=2;i<=n;++i)
		if(dep[i]>dep[x]) x=i;
	dfs(x,0);
	int y=1;
	for(int i=2;i<=n;++i)
		if(dep[i]>dep[y]) y=i;
	dfs1(x,0,y);
	dfs2(1,0);
	println(2*(n-1)-(ans+dep[y]-1)+2);
}
}

int main(){
	read(n),read(k);
	for(int i=1;i<n;++i){int u,v;read(u),read(v);add_edge(u,v),add_edge(v,u);}
	if(k==1) sub1::solve();
	else sub2::solve();
	return 0;
}