P2872Building Roads S

一、题目描述

　　

 

 二、解题思路

　　首先把这些点的距离预处理出来，然后把给定的边的权值设置为0。那么这些给定的边一定会被走到，然后根据短的再进行选就行了，因此可以用prim算法来做这个题，当然可以用堆优化，但是数据量比较小，所以就没有用堆优化了。（这里很容易出错，debug两个小时在一个double型的mn上，我哭了。。。）

三、代码实现

#include "bits/stdc++.h"
using namespace std;
int n,m;
int a[1010];
struct node{
    double x;
    double y;
}c[1010];
double graph[1010][1010];
double dis[1010];
int vis[1010];
const int inf = 1e9;
void prim()
{
    vis[1] = 1;
    for(int i = 1;i <= n;i++)
        dis[i] = graph[1][i];
    double mn;
    int k;
    double ans = 0;
    for(int i = 2;i <= n;i++){
        mn = inf;
        k = -1;
        for(int j = 1;j <= n;j++){
            if(!vis[j] && dis[j] < mn)
                mn = dis[j],k = j;
        }
        ans += mn;
        vis[k] = 1;
        for(int j = 1;j <= n;j++)
            if(!vis[j] && dis[j] > graph[k][j])
                dis[j] = graph[k][j];
    }
    printf("%.2lf",ans);
}
int main()
{
    cin >> n >> m;
    for(int i = 1;i <= n;i++)
        dis[i]  = inf;
    for(int i = 1;i <= n;i++){
        cin >> c[i].x >> c[i].y;
        for(int j = 1;j < i;j++){
            graph[i][j] = sqrt(pow(double(c[i].x - c[j].x),2) + pow(double(c[i].y - c[j].y),2));
            graph[j][i] = sqrt(pow(double(c[i].x - c[j].x),2) + pow(double(c[i].y - c[j].y),2));
        }
    }
    for(int i = 1;i <= m;i++){
        int u,v;
        cin >> u >> v;
        graph[u][v] = 0;
        graph[v][u] = 0;
    }
    prim();
    return 0;
}