[bzoj4883][Lydsy2017年5月月赛]棋盘上的守卫
来自FallDream的博客,未经允许,请勿转载, 谢谢。
在一个n*m的棋盘上要放置若干个守卫。对于n行来说,每行必须恰好放置一个横向守卫;同理对于m列来说,每列
必须恰好放置一个纵向守卫。每个位置放置守卫的代价是不一样的,且每个位置最多只能放置一个守卫,一个守卫
不能同时兼顾行列的防御。请计算控制整个棋盘的最小代价。
n*m<=10^5
费用流比较好想,把行和列拿出来,第i行向第j列连费用是a[i][j]的边,然后限制每行每列流量1即可。
但是费用流不是很科学(好像有人大力艹过了?),考虑优化。
费用流每次的增广路其实就是选择了费用最小的一行一列,假如把a[i][j]看作i->j的边,那么得到的显然会是一个环套树森林。
那么就跑最小生成树,然后记录每个点所在连通块是树还是图即可。
假如要合并i,j
如果ij都是图了,那么就没办法咯。
不然,i,j在同一个集合时加入这条边即可 树->图 ; 不在同一个集合的话,就把他们并起来,然后判断得到的是一个什么图形。
如果原来是树+树,得到树,树+图得到图。
复杂度nmlog(nm)
#include<iostream> #include<cstdio> #include<algorithm> #define rint register int #define getchar() (*S++) #define MN 100000 #define ll long long char B[1<<26],*S=B; using namespace std; inline int 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 * 10 + ch - '0';ch = getchar();} return x * f; } int n,m,cnt=0,fa[MN*2+5]; ll ans=0; struct data{int x,c,r;}a[MN+5]; bool b[MN*2+5]; bool cmp(data a,data b){return a.x<b.x;} inline int getfa(int x){return !fa[x]?x:fa[x]=getfa(fa[x]);} int main() { fread(B,1,1<<26,stdin); n=read();m=read(); for(rint i=1;i<=n;++i) for(rint j=1;j<=m;++j) a[++cnt]=(data){read(),i,j+n}; sort(a+1,a+cnt+1,cmp); for(rint i=1;i<=cnt;++i) { int x=getfa(a[i].c),y=getfa(a[i].r); if(!(b[x]&b[y])) { if(x!=y) fa[x]=y,b[y]|=b[x]; else b[x]=1; ans+=a[i].x; } } printf("%lld\n",ans); return 0; }
FallDream代表秋之国向您问好!
欢迎您来我的博客www.cnblogs.com/FallDream