原题直通车:HDU 3721 Building Roads
题意:从一棵树中去掉一条边,再建一条边将两棵树并成一棵,使得树中的最长路达到最小。
分析:有n-1条边,枚举去掉的边(长dis),求出两棵树中的最长路(长L); 再分别求两棵树
上的一点,满足树中其他点到它的最大距离最小(最小距离为Ma, Mb)。在这两点间建
立新边,则整棵树中通过新边的最长路的长边S=Ma+Mb+dis,那边新树中的最长路为max(L,S)。
最后求出整体的最小值!!
代码:
#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#define max(a,b) (a)>(b)?(a):(b)
#define min(a,b) (a)<(b)?(a):(b)
using namespace std;
const int maxn=2505;
const int inf=0XFFFFFFF;
struct node {
int to, dix, up;
node* next;
void read(int a,int b,node* c) {
to=a, dix=b, next=c, up=0;
}
}T[maxn<<1], *head[maxn];
int dp[maxn], fa[maxn], q[maxn], ptr, num;
void AddEdge(int a,int b,int c) {
T[ptr].read(b,c,head[a]);
head[a]=&T[ptr++];
}
void dfs_fa(int cnt) { /// 确定每个点的父结点
node* p=head[cnt];
while(p!=NULL) {
if(fa[p->to]==p->to&&p->to) {
fa[p->to]=cnt; dfs_fa(p->to);
}
p=p->next;
}
}
void dfs_up(int cnt) { ///求最长路的初始化
q[num++]=cnt; ///属于树中的结点
node* p=head[cnt];
while(p!=NULL) {
if(fa[p->to]==cnt) {
dfs_up(p->to);
dp[cnt]=max(dp[cnt], dp[p->to]+p->dix);
p->up=dp[p->to]+p->dix;
}
p=p->next;
}
}
void dfs_len(int cnt) { ///求树上的最长路
int pre=fa[cnt], mx=0;
node* p=head[pre];
while(p!=NULL) {
///求与fa[cnt]相邻的点(不包括cnt本身)到达fa[cnt]的最远距离
if(p->to!=cnt&&(fa[p->to]==pre||p->to==fa[pre]))
mx=max(mx, p->up);
p=p->next;
}
p=head[cnt];
while(p!=NULL) { ///更新经过cnt到fa[cnt]的最远距离
if(p->to==pre) {
p->up=mx+p->dix; break;
}
p=p->next;
}
p=head[cnt];
while(p!=NULL) {
if(fa[p->to]==cnt||p->to==fa[cnt]) {
dp[cnt]=max(dp[cnt], p->up);
if(fa[p->to]==cnt) dfs_len(p->to);
}
p=p->next;
}
}
void work(int root,int &ret,int &mx) {
num=0; /// 初始化树的结点为空
memset(dp, 0, sizeof(dp));
dfs_up(root);
node* p=head[root];
while(p!=NULL) {
if(fa[p->to]==root) dfs_len(p->to);
p=p->next;
}
ret=inf;
for(int i=0; i<num; ++i)
ret=min(ret, dp[q[i]]), mx=max(mx,dp[q[i]]);
}
int distan(int cnt) { /// 求cnt到其父亲结点的距离
node* p=head[cnt];
while(p!=NULL) {
if(p->to==fa[cnt]) return p->dix;
p=p->next;
}
}
int main() {
int T, cas=1; scanf("%d",&T);
while(T--) {
int n; scanf("%d",&n);
ptr=1; ///所边的数量
for(int i=0; i<n; ++i) fa[i]=i, head[i]=NULL;
for(int i=1; i<n; ++i) {
int a,b,c; scanf("%d%d%d",&a,&b,&c);
AddEdge(a,b,c); AddEdge(b,a,c);
}
dfs_fa(0);
int ans=inf,pre,Ma,Mb,dis,mx;
for(int i=1; i<n; ++i) {
dis=distan(i);
pre=fa[i]; mx=0; fa[i]=i; /// 分成两棵树,树根分别是0, i (即将i与fa[i]的关系断开)
work(0,Ma,mx); work(i,Mb,mx);
ans=min(ans, max(Ma+Mb+dis,mx));
fa[i]=pre;
}
printf("Case %d: %d\n",cas++, ans);
}
return 0;
}