poj 1751 prim

题目是说给了城镇的一些坐标，然后一些城镇是连通的，所以让连通的这些点的距离为0，然后用一个数组记录前驱，prim就可以求出来了，注意自己的答案不一定和上面的一样，因为是special judge，加油啊！~~~

#include <iostream>

#include <stdio.h>

using namespace std;

const int maxn=755;

const int maxm=1005;

const int inf=100000005;

int n,q;

struct node1

{

      int x;

      int y;

}map[maxn];

int dist[maxn];

int m[maxn][maxn];

int co[maxn][maxn];

int pre[maxm];

void prim()

{

      bool visit[maxn];

      int k;

      int minn;

      for(int i=1;i<=n;i++)

      {

            dist[i]=m[1][i];

            visit[i]=false;

      }

      visit[1]=true;

      dist[1]=0;

      for(int i=1;i<=n-1;i++)

      {

            minn=inf;

            k=0;

            for(int j=1;j<=n;j++)

            {

                  if(visit[j]==false&&dist[j]<minn)

                  {

                        minn=dist[j];

                        k=j;

                  }

            }

            if(minn!=0)printf("%d %d\n",pre[k],k);

            if(k==0)return;

            visit[k]=true;

            for(int j=1;j<=n;j++)

            {

                  if(visit[j]==false&&dist[j]>m[k][j]&&m[k][j]!=inf)

                  {

                        dist[j]=m[k][j];

                        pre[j]=k;

                  }

            }

      }

}

int main()

{

      freopen("in","r",stdin);

      scanf("%d",&n);

      for(int i=1;i<=n;i++)

      {

            scanf("%d%d",&map[i].x,&map[i].y);

      }

      for(int i=1;i<=n;i++)

      {

            for(int j=1;j<=n;j++)

            {

                  if(i!=j)m[i][j]=inf;

                  else m[i][j]=0;

            }

      }

      for(int i=1;i<=maxm;i++)

      {

            pre[i]=1;

      }

      for(int i=1;i<=n;i++)

      {

            for(int j=1;j<=i;j++)

            {

                  if(i!=j)

                  {

                        m[i][j]=m[j][i]=(map[i].x-map[j].x)*(map[i].x-map[j].x)+(map[i].y-map[j].y)*(map[i].y-map[j].y);

                  }

            }

      }

      scanf("%d",&q);

      for(int i=1;i<=q;i++)

      {

            int a,b;

            scanf("%d%d",&a,&b);

            m[a][b]=m[b][a]=0;

      }

      prim();

      return 0;

}