2020hdu多校第二场1001Total Eclipse

题目地址：http://acm.hdu.edu.cn/showproblem.php?pid=6763

题目大意：有一个n个点m条边的无向图，结点从1开始编号到n,每个结点都有一个权值b_i，现在希望可以将所有结点的权值改为0

可以进行下面的操作：

1：选择一个正整数k(1<=k<=n)

2：选择k个不同的结点c1,c2....ck(1<=ck<=n),并且这k个结点可以相互连通，也就是这k个点是连通的

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int N=100000+5;
const int M=200000+5;
int n,m,u,v;
struct node{
    int b,pos;
}pb[N];
int now[N];
vector<int>E[N];
int fa[N],p[N];
long long int s=0;
int cmp(struct node pb1,struct node pb2){
    return pb1.b>pb2.b;
}
int find(int x){//并查集查找祖宗节点同时进行路径压缩 
    if(fa[x]==x) return x;
    return fa[x]=find(fa[x]);
}
int add(int x,int y){
    int ry=find(y);//找到需要操作的集合的祖宗节点 
    s+=now[ry]-now[x];//贡献当前节点的操作权值 
    fa[ry]=x;//更改祖宗节点 
}
int main(){
    int t;
    cin>>t;
    while(t--){
        s=0;
        scanf("%d%d",&n,&m);
        //输入数据、初始化标志数组、初始化并查集数组
        //数据结构pb.i存储下标，pb.b存储该下标对应的权值b[i]
        //now[i]存储下标i对应的权值，因为上面的pb需要进行排序，并不能通过数组pb的下标i找到对应的b
        //,只能循环遍历查找，这样复杂度就会比较高 
        for(int i=1;i<=n;i++) scanf("%d",&pb[i].b),pb[i].pos=i,p[i]=0,fa[i]=i,now[i]=pb[i].b;
        while(m--){
            scanf("%d%d",&u,&v);
            E[u].push_back(v);//使用不定长数组存储图 
            E[v].push_back(u);
        }
        sort(pb+1,pb+n+1,cmp);//按照权值b从大到小进行排序 
        for(int i=1;i<=n;i++){//对排序好的数组，从最大的权值开始进行操作 
            u=pb[i].pos;//找到未访问的最大权值的所对应的下标 
            int size=E[u].size();//该下标所对应的节点的边数
            p[u]=1;//对于放入图中的点进行标记 
            for(int j=0;j<size;j++){
                if(p[E[u][j]]==0) continue;//只有当与当前节点所连的节点已经放入图中时，才会进行下一次操作 
                //若是跟u相连的其他节点已经放入图中，并且还是在一个集合中，则不需要再此进行操作 
                //其实不需要这个判定条件也可以，因为即使其他所连节点在一个集合中也没有关系
                //因为当第二次执行这个函数时（第二次对之前已经做过的集合进行操作时），之前的集合的祖宗节点已经是u
                //所以对于s而言是相当于加了一个0，对于最后的结果并没有什么影响 
                if(find(u)!=find(E[u][j])) 
                    add(u,E[u][j]);//将u放如途中所需要进行的函数操作 
            }
        }
        for(int i=1;i<=n;i++)
            if(fa[i]==i){
                s+=now[i];//最后还需要将所有的祖宗节点的权值加入进s 
            }
        cout<<s<<endl;
        for(int i=1;i<=n;i++) E[i].clear();//这一步要注意，之前没有这一步显示栈溢出 
    }
    return 0;
}

2020hdu多校第二场1001Total Eclipse

猜你喜欢