350. 巡逻

在一个地区有 n 个村庄，编号为1,2,…,n。

有 n-1 条道路连接着这些村庄，每条道路刚好连接两个村庄，从任何一个村庄，都可以通过这些道路到达其他任一个村庄。

每条道路的长度均为1个单位。

为保证该地区的安全，巡警车每天都要到所有的道路上巡逻。

警察局设在编号为1的村庄里，每天巡警车总是从警局出发，最终又回到警局。

为了减少总的巡逻距离，该地区准备在这些村庄之间建立 K 条新的道路，每条新道路可以连接任意两个村庄。

两条新道路可以在同一个村庄会合或结束，甚至新道路可以是一个环。

因为资金有限，所以 K 只能为1或2。

同时，为了不浪费资金，每天巡警车必须经过新建的道路正好一次。

编写一个程序，在给定村庄间道路信息和需要新建的道路数的情况下，计算出最佳的新建道路的方案，使得总的巡逻距离最小。
输入格式

第一行包含两个整数 n 和 K。

接下来 n-1 行每行两个整数 a 和 b，表示村庄 a 和 b 之间有一条道路。
输出格式

输出一个整数，表示新建了 K 条道路后能达到的最小巡逻距离。
数据范围

3≤n≤100000
,
1≤K≤2,
1≤a,b≤n

输入样例：

8 1
1 2
3 1
3 4
5 3
7 5
8 5
5 6

输出样例：

11
注意：
dfs求直径只能在正边权上求，若取反产生负边权只能用树形dp

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
#define N 100005
using namespace std;
int head[N],ver[N<<1],nex[N<<1],edge[N<<1];
int n,k,tot=1,fa[N],pre_pos,pos,maxn,pre[N],v[N],f[N];

inline void add(int x,int y,int z){
	nex[++tot]=head[x];
	head[x]=tot;ver[tot]=y;
	edge[tot]=z;
}
void dfs(int x,int val){
	if(val>maxn){
		pos=x;maxn=val;
	}
	for(int i=head[x];i;i=nex[i]){
		int y=ver[i];
		if(y==fa[x])continue;
		
		fa[y]=x;pre[y]=i;
		dfs(y,val+edge[i]);
	}
}
void reset(int x){
	if(pre[x]==0)return ;
	edge[pre[x]^1]=edge[pre[x]]=-1;
	reset(ver[pre[x]^1]);
}
void dp(int x) {
        v[x] = 1;
        for (int i = head[x]; i; i = nex[i])
            if (!v[ver[i]]) {
                dp(ver[i]);
                maxn = max(maxn, f[ver[i]] + f[x] + edge[i]);
                f[x] = max(f[x], f[ver[i]] + edge[i]);
            }
    }
int main()
{
    cin>>n>>k;
    for(int i=1;i<n;i++){
	    int x,y,z;
		cin>>x>>y;z=1;
		add(x,y,z);add(y,x,z);
	}
	maxn=0;
	fa[1]=-1;
	dfs(1,0);
	maxn=0;memset(fa,0,sizeof(fa));
	pre_pos=pos;
	memset(pre,0,sizeof(pre));
	dfs(pos,0);
	long long ans=2*(n-1)-maxn+1;
	long long l=maxn;
	if(k==1){
		cout<<ans<<endl;
		return 0;
	}
	
	reset(pos);
	maxn=0;
	dp(1);
//	if(n==80000&&k==2)cout<<maxn<<endl;
	ans=min(ans,2*n-l-maxn);
	cout<<ans<<endl;
	return 0;
}
/*
146533 
146479
*/
/*
146533
146519
*/