[BZOJ4883][Lydsy1705月赛]棋盘上的守卫(Kruskal)

对每行每列分别建一个点,问题转为选n+m条边,并给每条边选一个点覆盖,使每个点都被覆盖。也就是最小生成环套树森林。

用和Kruskal一样的方法,将边从小到大排序,若一条边被选入后连通块仍然是一个环套树(即边数不多于点数)则连上。证明大致同Kruskal。

 1 #include<cstdio>
 2 #include<algorithm>
 3 #define rep(i,l,r) for (int i=(l); i<=(r); i++)
 4 typedef long long ll;
 5 using namespace std;
 6 
 7 const int N=100010;
 8 ll ans;
 9 int n,m,x,tot,fa[N],cnt[N],sz[N];
10 struct E{ int x,y,z; }e[N];
11 bool operator <(const E &a,const E &b){ return a.z<b.z; }
12 
13 int get(int x){ return (fa[x]==x) ? x : fa[x]=get(fa[x]); }
14 void uni(int x,int y){ fa[x]=y; sz[y]+=sz[x]; cnt[y]+=cnt[x]+1; }
15 
16 int main(){
17     freopen("bzoj4883.in","r",stdin);
18     freopen("bzoj4883.out","w",stdout);
19     scanf("%d%d",&n,&m);
20     rep(i,1,n) rep(j,1,m) scanf("%d",&x),e[++tot]=(E){i,j+n,x};
21     sort(e+1,e+tot+1);
22     rep(i,1,n+m) fa[i]=i,sz[i]=1;
23     rep(i,1,n*m){
24         int x=get(e[i].x),y=get(e[i].y);
25         if (x!=y) { if (cnt[x]+cnt[y]+1<=sz[x]+sz[y]) ans+=e[i].z,uni(x,y); }
26             else if (cnt[x]<sz[x]) ans+=e[i].z,cnt[x]++;
27     }
28     printf("%lld
",ans);
29     return 0;
30 }
原文地址:https://www.cnblogs.com/HocRiser/p/10126916.html