P1550 [USACO08OCT]打井Watering Hole
阿新 • • 發佈:2018-10-26
www. 過去 pty pri line ble code lin hole
傳送門
如果沒有挖井只是聯通所有點就是最小生成樹
但是有挖井,我們需要轉換一下
把挖井操作看成 0 號點連一條邊過去
然後還是最小生成樹..
因為是稠密圖,所以用 prim 算法來求最小生成樹
#include<iostream> #include<cstdio> #include<cstring> #include<cmath> #include<cstdio> #include<queue> using namespace std; typedef long long ll; inlineint read() { int x=0,f=1; char ch=getchar(); while(ch<‘0‘||ch>‘9‘) { if(ch==‘-‘) f=-1; ch=getchar(); } while(ch>=‘0‘&&ch<=‘9‘) { x=(x<<1)+(x<<3)+(ch^48); ch=getchar(); } return x*f; } const int N=507,INF=0x3f3f3f3f; int n,ans,mp[N][N]; struct data {int dis,id; inline bool operator < (const data &tmp) const { return dis>tmp.dis; } }; priority_queue <data> q; int dis[N]; bool vis[N]; void prim(int st)//prim模板 { memset(dis,INF,sizeof(dis)); q.push((data){0,st}); dis[0]=0; while(!q.empty()) { data x=q.top(); q.pop(); if(vis[x.id]) continue; vis[x.id]=1; ans+=x.dis; for(int i=0;i<=n;i++) if(dis[i]>mp[x.id][i]) { dis[i]=mp[x.id][i]; q.push((data){mp[x.id][i],i}); } } } int main() { n=read(); for(int i=1;i<=n;i++) mp[0][i]=mp[i][0]=read();//0號點連邊 for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) mp[i][j]=mp[j][i]=read(); prim(0); printf("%d",ans); return 0; }
P1550 [USACO08OCT]打井Watering Hole