[APIO2010]巡逻--树的直径bfs和dp求法

Luogu 3629

题目分析：

• 分析可知，加一条边会形成环，我们在树的直径两端点加一条，距离减得最多

• 当要加两条边时，如果两条边不重叠

• 如果重叠，考虑两条边都必须走一次，所以我们可以在第一次找出直径后，将路径上的权值变为-1，若边重叠，则减去$-1$之后相当于加一

• 所以:
  $if(m==1)ans=2\ast (n-1)-le{n}_1+1$
  $if(m==2)ans=2\ast (n-1)-le{n}_1+1-le{n}_2+1$

• $bfs$求直径易记录路径，但无法处理负权边，所以第二次求直径不能用$bfs$

• $dp$求直径不易记录路径，可以处理负权边

Code:

#include <bits/stdc++.h>
using namespace std;
#define maxn 100010
#define INF 1061109567

int f[maxn],maxd=0,p,q,d[maxn],pre[maxn],size=1,n,m,head[maxn];
struct edge {
	int v,nxt,w;
}e[maxn<<1];

inline void init_() {
	freopen("a.txt","r",stdin);
}

inline int read_() {
	int x=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9') {
		if(c=='-') f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9') {
		x=(x<<3)+(x<<1)+c-'0';
		c=getchar();
	}
	return x*f;
}

inline void clean_() {
	memset(head,-1,sizeof(head));
	memset(f,0,sizeof(f));
}

inline void add_(int u,int v,int w) {
	e[++size].v=v;
	e[size].w=w;
	e[size].nxt=head[u];
	head[u]=size;
}

inline int bfs_(int s) {
	memset(d,0x3f,sizeof(d));
	queue<int > q;
	q.push(s);
	d[s]=0;
	while(!q.empty()) {
		int u=q.front();
		q.pop();
		for(int i=head[u];~i;i=e[i].nxt) {
			int v=e[i].v;
			if(d[v]==INF) {
				d[v]=d[u]+e[i].w;
				pre[v]=u;
				q.push(v);
			}
		}
	}
	int maxd=0,max_u=0;
	for(int i=1;i<=n;++i) if(maxd<d[i]) maxd=d[i],max_u=i;
	return max_u;
} 

int get_() {
	p=bfs_(1);
	q=bfs_(p);
	return d[q];
}

inline void change_() {
	if(m==2) {
		for(int i=q;i!=p;i=pre[i]) {
			for(int j=head[i];~j;j=e[j].nxt) {
				if(e[j].v==pre[i]) {
					e[j].w=e[j^1].w=-1;
				}
			}
		}
	}
}

void dp_(int u,int fa) {
	for(int i=head[u];~i;i=e[i].nxt) {
		int v=e[i].v;
		if(v==fa) continue;
		dp_(v,u);
		maxd=max(maxd,f[u]+f[v]+e[i].w);
		f[u]=max(f[u],f[v]+e[i].w);
	}
}


void readda_() {
	n=read_();m=read_();
	clean_();
	int x,y,z;
	for(int i=1;i<n;++i) {
		x=read_();y=read_();
		add_(x,y,1);add_(y,x,1);
	}
	x=get_();z=1;
	if(m==2) {
		change_();
		dp_(1,0);
		z=2;
	}
	printf("%d",((n-1)<<1)-x-maxd+z);
}

int main() {
	init_();
	readda_();
	return 0;
}