最小生成树prim算法

prim算法是一种用贪心思想的最小生成树算法。

初始先找到一个点，之后从这个点往下找，找一个权值最小的边及其到达的点，把这个点和第一个点当成一个生成树，再重复之前的步骤，最后找到n-1条边则停止。

Prim在稠密图中比Kruskal优，在稀疏图中比Kruskal劣。

例题：洛谷P2212 [USACO14MAR]Watering the Fields S

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<vector>
#include<list>
#include<queue>
#include<stack>
#include<set>
using namespace std;
typedef long long ll;
ll n,c;
const int inf=0x3f3f3f3f;
ll dis[5001];
bool vis[5001];
ll x[100000],y[100000];
ll calc(ll u,ll v){
    ll r=(x[u]-x[v])*(x[u]-x[v])+(y[u]-y[v])*(y[u]-y[v]);
    if(r<c){
        return inf;
    }
    return r;
}
void prim(){
    for(int i=1;i<=n;i++){
        dis[i]=inf;
    }
    dis[1]=0;
    for(int i=1;i<=n;i++){
        ll p=0;
        for(int j=1;j<=n;j++){
            if(!vis[j]){
                if(!p || dis[j]<dis[p]){
                    p=j;
                }
            }
        }
        vis[p]=1;
        for(int i=1;i<=n;i++){
            if(!vis[i]){
                dis[i]=min(dis[i],calc(p,i));
            }
        }
    }
    ll ans=0;
    for(int i=2;i<=n;i++){
        if(dis[i]==inf){
            cout<<"-1"<<endl;
            return;
        }
        ans+=dis[i];
    }
    cout<<ans;
} 
int main(){
    cin>>n>>c;
    for(int i=1;i<=n;i++){
        cin>>x[i]>>y[i];
    }
    prim();
    return 0;
}