-
Highways
- POJ - 1751
- 记录一下当前点被谁更新建立pre数组记录就解决了
-
#include<algorithm> #include<iostream> #include<cmath> #include<iomanip> using namespace std; #define maxn 777 #define inf 0x3f3f3f3f double mmp[maxn][maxn]; double dis[maxn],ans[maxn]; int n,m,t,u,v; double x[maxn],y[maxn]; bool vis[maxn]; void prim() { dis[1]=0; for(int i=0; i<n; i++) { double temp=inf; int k=-1; for(int j=1; j<=n; j++) if(dis[j]<temp&&!vis[j]) temp=dis[k=j]; if(k==-1) break; vis[k]=1; if(i>0&&dis[k]) cout<<ans[k]<<" "<<k<<endl; for(int j=1; j<=n; j++) if(dis[j]>mmp[k][j]&&!vis[j]) { dis[j]=mmp[k][j]; ans[j]=k; } } return ; } double cal(int i,int j) { return sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])); } int main() { cin>>n; for(int i=1; i<=n; i++) { cin>>x[i]>>y[i]; dis[i]=inf; vis[i]=0; } for(int i=1; i<=n; i++) for(int j=1; j<=n; j++) mmp[i][j]=mmp[j][i]=cal(i,j); cin>>m; while(m--) { cin>>u>>v; mmp[u][v]=mmp[v][u]=0; } prim(); return 0; }
Highways -PRIM打印生成树边
最新推荐文章于 2022-02-25 15:43:19 发布