P3629 [APIO2010]巡逻 （树的直径 思维）

[APIO2010]巡逻 - 洛谷

感觉有点偏思维把，但是仔细想想还是可做的。

首先每条路都要走，那么我们把最长的路首位连接，就可以少最长的路的距离-1.因为不用来回走了。即求直径。

那么如果是两条路，我们也要求另外一条直径。但是此时，第一条和第二条中重合的部分就不能算了，算就是只走一次的意思，但重合导致我们要走两遍，所以反而还要把边权变负。那么再求一次直径即可。

数据比较好搞。前一次可以用dfs，顺便求出直径信息；

后一次有负边，只能dp求。dp好像也能求直径信息吧，不会啊。

/*keep on going and never give up*/
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define ll long long
#define db(x) cerr<<(#x)<<" "<<(x)<<" "<<endl;
#define endl "\n"
#define fast std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const double E = exp(1);
const double PI = acos(-1.0);
const int mod=1e9+7; 
const int maxn=2e5+10;
int n,m,k;
struct node{
	int to,w,nxt;
}e[maxn<<1];int tot,head[maxn<<1];
void add(int x,int y,int z){ e[++tot]={y,z,head[x]},head[x]=tot;}
int vis[maxn],dis[maxn],fa[maxn],d[maxn],l1,l2;
void dfs(int x){
	for(int i=head[x];i;i=e[i].nxt){
		int v=e[i].to,w=e[i].w;
		if(v==fa[x]) continue;
		fa[v]=x;
		dis[v]=dis[x]+w;
		dfs(v);
	}
}
void dp(int x,int fa){
	for(int i=head[x];i;i=e[i].nxt){
		int v=e[i].to;
		if(v==fa) continue;
		if(vis[x]&&vis[v]) e[i].w=-1;
		dp(v,x);
		l1=max(l1,d[x]+d[v]+e[i].w);
		d[x]=max(d[x],d[v]+e[i].w);
	}
}
signed main(){
	fast
	cin>>n>>k;int x,y;
	for(int i=1;i<n;i++){
		cin>>x>>y;
		add(x,y,1);add(y,x,1);
	}
	int be=1,en=1;
	dis[be]=0;dfs(be);
	memset(fa,0,sizeof(fa));
	for(int i=1;i<=n;i++) if(dis[i]>dis[be]) be=i;
	dis[be]=0;dfs(be);
	for(int i=1;i<=n;i++) if(dis[i]>dis[en]) en=i;
	l2=dis[en];
	for(int i=en;i;i=fa[i]) vis[i]=1;
	if(k==1){
		cout<<2*n-l2-1;return 0;
	}
	dp(1,0);
//	cout<<l2<<" "<<l1<<endl;
	cout<<2*n-l2-l1;
}