【BZOJ 1601】灌水(设置虚点+MST)

最新推荐文章于 2018-03-20 19:59:00 发布

Etta19

最新推荐文章于 2018-03-20 19:59:00 发布

阅读量318

点赞数

分类专栏：图论 MST

本文链接：https://blog.csdn.net/Etta19/article/details/77367955

版权

图论同时被 2 个专栏收录

20 篇文章 0 订阅

订阅专栏

MST

3 篇文章 0 订阅

订阅专栏

传送门

祝不被权限QvQ
题意：n个点，给出两两点修路引水费用与就地建水库费用，求使所有点直接或间接得到水的最小花费。

I think

设置虚点连接建水库的点 ,建水库费用即该点与虚点的边权,添加完其他边Kruskal跑MST。

Code

#include<cstdio>
#include<algorithm>
using namespace std;

const int sm = 300+10;

int x,N,Ct,Ans;
int Fa[sm];
struct Ed{
    int Fm,To,L;
    bool operator < (const Ed&a) const {
        return L<a.L;
    }
}e[sm*(sm+1)>>1];

void Ins(int f,int t,int l) { e[++Ct]=(Ed){f,t,l}; }
int Find(int x) { return x==Fa[x]?x:Fa[x]=Find(Fa[x]); }
int main() {
    freopen("1.in","r",stdin);
    scanf("%d",&N);
    for(int i=1;i<=N;++i)
        scanf("%d",&x),Ins(0,i,x),Fa[i]=i;
    for(int i=1;i<=N;++i)
        for(int j=1;j<=N;++j) {
            scanf("%d",&x); 
            if(j>i) Ins(i,j,x);
        }
    sort(e+1,e+Ct+1);
    for(int i=1,x,y,z=0;i<=Ct;++i) {
        x=Find(e[i].Fm);
        y=Find(e[i].To);
        if(x!=y) {
            Fa[x]=y,Ans+=e[i].L;
            ++z; if(z==N) break;
        }
    }
    printf("%d\n",Ans);
    return 0;
}