[BZOJ1912]APIO2010巡逻|树的直径

    这题看完大概的做法也想到了。。就是找出两条最长的路径连成环嘛。。但是不能简单的这样考虑,因为找次长的时候并不是不能走最长的路径上的边,只是走了第一次就等于白走了,也就是不仅没有产生新的贡献反而抵消了第一次的贡献。。根据这个思路,K=1的时候直接找出一条最长路径len,答案即为2*(n-1)-len+1K=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);
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值