bzoj1601 / P1550 [USACO08OCT]打井Watering Hole(堆优化prim)

 
对于自己建水库的情况,新建一个虚拟结点,和其他点的边权即为自建水库的费用
这样问题就转化为一个裸最小生成树问题了。
这里用堆优化prim解决。
 #include<iostream>
#include<cstdio>
#include<cstring>
#include<cctype>
#include<queue>
#define re register
using namespace std;
void read(int &x){
char c=getchar();x=;
while(!isdigit(c)) c=getchar();
while(isdigit(c)) x=(x<<)+(x<<)+(c^),c=getchar();
}
struct data{
int d,u;
data(){}
data(int A,int B):
d(A),u(B)
{}
bool operator < (const data &tmp) const{return d>tmp.d;}
};priority_queue <data> h;
#define N 305
int n,a[N][N],d[N],ans; bool vis[N];
int main(){
read(n);++n; int k=;
for(re int i=;i<n;++i)
read(a[n][i]),a[i][n]=a[n][i];//新建边
for(re int i=;i<n;++i)
for(re int j=;j<n;++j)
read(a[i][j]);
memset(d,,sizeof(d));
h.push(data(d[]=,));
while(!h.empty()&&k<n){
data x=h.top(); h.pop();
if(vis[x.u]) continue;
vis[x.u]=; ++k; ans+=x.d;
for(re int i=;i<=n;++i)
if(a[x.u][i]){
if(d[i]>a[x.u][i]){
d[i]=a[x.u][i];
h.push(data(a[x.u][i],i));
}
}
}printf("%d",ans);
return ;
}
上一篇:P3366 【模板】最小生成树(堆优化prim)


下一篇:堆优化Prim 最小生成树 模板