题目链接
题目大意
n个点的图,每个点有点权,每次选取一个最大联通分量,然后其中每个点点权-1,点权不能为负,求最少需要操作的次数
题解
按照点权对所有点进行排序,然后依次加边建图,进行反向并查集。先将答案加上当前点的点权,如果当前点(s)连出去的点(t)之前已经访问过,说明在访问t点的时候多加了一次s的点权,那么将答案减去一次s点的点权。
标程
#include<bits/stdc++.h>
using namespace std;
const long long maxn=1e5+5;
long long fa[maxn];
bool vis[maxn];
struct node
{
long long idx,val;
}N[maxn];
bool cmp(node &A,node &B)
{
return A.val>B.val;
}
long long find(long long x)
{
return fa[x]==x ? x:fa[x]=find(fa[x]);
}
vector<long long > E[maxn];
int main()
{
long long t;
scanf("%d",&t);
while (t--)
{
long long n,m;
cin>>n>>m;
for (long long i = 1; i <= n; i++)
{
E[i].clear();
scanf("%lld",&N[i].val);
N[i].idx=i;
fa[i]=i;
}
sort(N+1,N+1+n,cmp);
while (m--)
{
long long x,y;
scanf("%lld%lld",&x,&y);
E[x].push_back(y);
E[y].push_back(x);
}
memset(vis,0,sizeof(vis));
long long ans=0;
for (long long i = 1; i <=n; i++)
{
ans+=N[i].val;
long long id=N[i].idx;
vis[id]=1;
long long fai=find(fa[id]);
for(long long j=0;j<E[id].size();j++)
{
long long x=E[id][j];
if(!vis[x]) continue;
long long fax=find(x);
if(fax==fai)
continue;
fa[fax]=fai;
ans-=N[i].val;
}
}
printf("%lld\n",ans);
}
return 0;
}