[BZOJ1912]APIO2010巡逻|树的直径

    这题看完大概的做法也想到了。。就是找出两条最长的路径连成环嘛。。但是不能简单的这样考虑，因为找次长的时候并不是不能走最长的路径上的边，只是走了第一次就等于白走了，也就是不仅没有产生新的贡献反而抵消了第一次的贡献。。根据这个思路，K=1的时候直接找出一条最长路径len，答案即为2*(n-1)-len+1，K=2时，据前面的分析，把所有最长路径上的边权改为-1（双向！！），然后再求最长，答案即为2*(n-1)-len1+1-len2+1。。求最长我一开始想的就是一遍DFS维护最长次长子路径就能搞，但是想起NOIP做过一道是直接两遍DFS,就做了两遍。。后来发现做两遍的方法当有负权边的时候是不可行的，然后就换成第一种了。。

#include<cstdio>
#include<iostream>
#include<memory.h>
#define N 100005
#define clr(a) memset(a,0,sizeof(a))
using namespace std;
struct edge{
	int e,q,next;
} ed[N*2];
int n,k,s,e,i,ne=1,len,ans,a[N],d[N],son[N][2];
void add(int s,int e)
{
	ed[++ne].e=e;ed[ne].q=1;
	ed[ne].next=a[s];a[s]=ne;
}
int dfs(int x,int fa)
{
	int j,to,k,max1=0,max2=0;
	son[x][0]=son[x][1]=-1;
	for (j=a[x];j;j=ed[j].next)
		if ((to=ed[j].e)!=fa)
		{
			d[to]=d[x]+ed[j].q;
			k=dfs(to,x)+ed[j].q;
			if (k>max1) 
			{
				son[x][1]=son[x][0];
				son[x][0]=j;
				max2=max1;
				max1=k;
				if (max1+max2>len) len=max1+max2,s=x;
			}
			else if (k>max2)
			{
				son[x][1]=j;
				max2=k;
				if (max1+max2>len) len=max1+max2,s=x;
			}
		}
	return max1;	
}
int main()
{
	freopen("1912.in","r",stdin);
//	freopen("my.out","w",stdout);
	scanf("%d%d",&n,&k);
	clr(a);
	for (i=1;i<n;i++)
	{
		scanf("%d%d",&s,&e);
		add(s,e);add(e,s);
	}
	d[1]=0;len=1;
	s=0;son[0][1]=son[0][0]=-1;
	dfs(1,0);
	ans=2*n-2;
	ans-=len-1;
	if (k==2)
	{
		for(i=son[s][0];i!=-1;i=son[ed[i].e][0]) ed[i].q=ed[i^1].q=-1;
		for(i=son[s][1];i!=-1;i=son[ed[i].e][0]) ed[i].q=ed[i^1].q=-1;
		d[e]=0;len=1;
		dfs(e,0);
		ans-=len-1;
	} 	
	printf("%d\n",ans);
}