题目大意:求MST,要求制定的一条边在MST中。输出该树的最小路径权值。
解题思路:这个题的描述很类似次小生成树的求法,直接进行对最小生成树的边进行标记以及生成树的每两个点之间最大边权值进行记录,最后判断这条边在没在MST中,在则直接输出,否则加入改变权值再减去这条边两个端点之间MST中权值最大的哪一条边即可,详见code。
题目来源:http://acm.hdu.edu.cn/showproblem.php?pid=4463
code:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int MAXN = 50+5;
const int inf = 0x3fffffff;
int n,q,p;
int pre[MAXN],vis[MAXN];
double dist[MAXN],g[MAXN][MAXN],path[MAXN][MAXN];
bool used[MAXN][MAXN];
struct point{
int x,y;
}a[MAXN];
double distan(point a,point b){
return sqrt(double(a.x-b.x)*(a.x-b.x)+double(a.y-b.y)*(a.y-b.y));
}
double prim(){
double ans=0;
memset(vis,0,sizeof(vis));
memset(path,0,sizeof(path));
memset(used,0,sizeof(used));
vis[1]=1;
for(int i=1;i<=n;++i){
dist[i]=g[1][i];
pre[i]=1;
}
for(int i=1;i<n;++i){
int u=-1;
for(int j=1;j<=n;++j)
if(!vis[j] && (u==-1 || dist[j]<dist[u])) u=j;
used[u][pre[u]]=used[pre[u]][u]=true;
ans+=g[pre[u]][u];
vis[u]=1;
for(int k=1;k<=n;++k){
if(vis[k] && k!=u)
path[u][k]=path[k][u]=max(dist[u],path[k][pre[u]]);
if(!vis[k] && dist[k]>g[u][k]){
dist[k]=g[u][k];
pre[k]=u;
}
}
}
return ans;
}
int main(){
//freopen("input.txt","r",stdin);
while(~scanf("%d",&n) && n){
scanf("%d%d",&p,&q);
for(int i=1;i<=n;++i)
scanf("%d%d",&a[i].x,&a[i].y);
for(int i=1;i<=n;++i)
for(int j=1;j<=n;++j)
if(i!=j) g[i][j]=distan(a[i],a[j]);
double mst=prim();
if(used[p][q]) printf("%.2f\n",mst);
else printf("%.2f\n",mst-path[p][q]+g[p][q]);
}
return 0;
}