bzoj 3479: [Usaco2014 Mar]Watering the Fields 最小生成树

→题目链接←

裸的最小生成树...只会prim算法

n^2暴力建边

然后跑最小生成树

学了这么长的时间终于第一次写最小生成树

真棒啊

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<queue>

using namespace std;

struct node{
int s,t;
int len;
node(int x,int y,int z){
s=x;
t=y;
len=z;
}
friend bool operator < (node a,node b){
return a.len>b.len;
}
};

struct point{
int x,y;
}p[2020];

priority_queue<node>q;
vector<int>v[2020];
bool vis[2020];
int n,m;

int cal(int x,int y){
return (p[x].x-p[y].x)*(p[x].x-p[y].x)+(p[x].y-p[y].y)*(p[x].y-p[y].y);
}

int main(){
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++){
int x,y;
scanf("%d%d",&p[i].x,&p[i].y);
}
for(int i=1; i<=n; i++){
for(int j=i+1; j<=n; j++){
if(cal(i,j)>=m){
v[i].push_back(j);
v[j].push_back(i);
}
}
}
int ans=0;
int sum=1;
vis[1]=true;
for(int i=0; i<v[1].size(); i++)q.push(node(1,v[1][i],cal(1,v[1][i])));
while(!q.empty()){
node e=q.top();
q.pop();
if(vis[e.t])continue;
ans+=e.len;
sum++;
if(sum==n)break;
vis[e.t]=true;
for(int i=0; i<v[e.t].size(); i++)q.push(node(e.t,v[e.t][i],cal(e.t,v[e.t][i])));
}
if(sum==n)printf("%d\n",ans);
else printf("-1\n");

return 0;
}