巡逻（树的直径 树形dp+两次dfs）

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+100;
int h[N],e[2*N],ne[2*N],v[2*N],idx;
int dis[N];
int pre[N];
bool st[N];
int d[N];
int ans=0;
int n,k;
int p;
void add(int a,int b){
	v[idx]=1,e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u){
	for(int i=h[u];i!=-1;i=ne[i]){
		int j=e[i];
		if(dis[j]!=0x3f3f3f3f) continue;
		if(dis[j]==0x3f3f3f3f){
			dis[j]=dis[u]+1;
			pre[j]=i;
		}
		dfs(j);
	}
}
void change(){
	int i=p;
	while(pre[i]!=-1){
		v[pre[i]]=-1;v[pre[i]^1]=-1;
		i=e[pre[i]^1];
	}
}
void dp(int x){
	st[x]=1;
	for(int i=h[x];i!=-1;i=ne[i]){
		int j=e[i];
		if(st[j]) continue;
		dp(j);
		ans=max(ans,d[x]+d[j]+v[i]);
		d[x]=max(d[x],d[j]+v[i]);
	}
}
int main(){
	memset(h,-1,sizeof h);
	scanf("%d%d",&n,&k);
	for(int i=0;i<n-1;i++){
		int a,b;scanf("%d%d",&a,&b);
		add(a,b);
		add(b,a);
	} 
	memset(dis,0x3f,sizeof dis);
	dis[1]=0;
	dfs(1);
	int g=-1;
	for(int i=1;i<=n;i++){
		if(dis[i]>g){
			g=dis[i];p=i;
		}
	}
	memset(dis,0x3f,sizeof dis);
	memset(pre,-1,sizeof pre);
	dis[p]=0;
	dfs(p);
	g=-1;
	for(int i=1;i<=n;i++){
		if(dis[i]>g){
			g=dis[i];p=i;
		}
	}
	int l1=dis[p];
	if(k==1){
		printf("%d\n",2*(n-1)-l1+1);
		return 0;
	}
	change();
	dp(1);
	int l2=ans;
	printf("%d\n",2*(n-1)-l1-l2+2);
	return 0;
}