Highways -PRIM打印生成树边

• 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;
  }
  

•