POJ- 1751 Highways

题目链接：http://poj.org/problem?id=1751

 

Sample Input
9
1 5
0 0 
3 2
4 5
5 1
0 4
5 2
1 2
5 3
3
1 3
9 7
1 2
Sample Output
1 6
3 7
4 9
5 7
8 3

 

分析：某个地方政府想修建一些高速公路使他们每个乡镇都可以相同通达，不过以前已经修建过一些公路，现在要实现所有的联通，所花费的最小代价是多少？（也就是最小的修建长度），输出的是需要修的路，不过如果不需要修建就什么都不输出。

一般的最小生成树，注意控制下标，输出来就好了

 

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<iostream>
#include<queue>
#include<stdlib.h>
using namespace std;

#define N 12000
#define INF 0x3f3f3f3f

double maps[N][N],dist[N],c[N];
int n,vis[N],v=0;

struct node
{
    int x,y;
} p[N],q[N];

void Init()
{
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
            maps[i][j]=(i==j)?0:INF;
}

double Len(node a,node b)
{
    int x=b.x-a.x;
    int y=b.y-a.y;
    double len=(double)sqrt(x*x+y*y);
    return len;
}

void Prim(int s)
{
    vis[s]=1;

    for(int i=1; i<=n; i++)
    {
        dist[i]=maps[s][i];
        q[i].x=s,q[i].y=i;
    }


    for(int i=1; i<n; i++)
    {
        double Min=INF;
        int index=-1;
        for(int j=1; j<=n; j++)
        {
            if(!vis[j]&&Min>dist[j])
            {
                Min=dist[j];
                index=j;
            }
        }
        if(Min !=0)
           printf("%d %d\n", q[index].x,q[index].y);

        vis[index]=1;

        for(int j=1; j<=n; j++)
            if(!vis[j]&&dist[j]>maps[index][j])
            {
                dist[j]=maps[index][j];
                q[j].x=index,q[j].y=j;
            }
    }
}

int main()
{
    int i,j,m,a,b;

    scanf("%d", &n);
    Init();

    for(i=1; i<=n; i++)
        scanf("%d%d", &p[i].x,&p[i].y);

    for(i=1; i<n; i++)
        for(j=i+1; j<=n; j++)
            maps[i][j]=maps[j][i]=Len(p[i],p[j]);


    scanf("%d", &m);

    for(i=1; i<=m; i++)
    {
        scanf("%d%d", &a,&b);
        maps[a][b]=maps[b][a]=0;
    }

    Prim(1);

    return 0;
}