/*
离线并查集,边按小到大排序,询问也按小到大排序;
对于询问x,答案就是询问x-1的值加上询问x-1的L1和询问x的L2之间的边合并带来的值
对于一条属于L1和L2之间的边,如果端点u,v在一个集合中,忽略,否则新开的路是u
所在集合的大小乘上v所在集合的大小
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<map>
using namespace std;
const int M=50009;
const int N=10009;
struct Edge
{
int u,v,w;
}edge[M];
bool cmp(Edge a,Edge b)
{
return a.w<b.w;
}
struct Que
{
int num,val;
}Qu[N];
bool cmp2(Que a,Que b)
{
return a.val<b.val;
}
int n,m,q;
int fa[N],heav[N];
int ans[N];
void init()
{
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
}
}
int find(int x)
{
return fa[x]==x?x:fa[x]=find(fa[x]);
}
int Union(int x,int y)
{
if(heav[x]>heav[y])
{
fa[y]=x;
heav[x]+=heav[y];
}
else
{
fa[x]=y;
heav[y]+=heav[x];
}
}
void solve()
{
for(int i=1;i<=n;i++)
{
fa[i]=i;
heav[i]=1;
}
for(int i=1;i<=q;i++)
{
scanf("%d",&Qu[i].val);
Qu[i].num=i;
}
sort(edge,edge+m,cmp);
sort(Qu+1,Qu+q+1,cmp2);
int j=0;
for(int i=1;i<=q;i++)
{
ans[Qu[i].num]=ans[Qu[i-1].num];
while(edge[j].w<=Qu[i].val&&j<m)
{
int x=find(edge[j].u);
int y=find(edge[j].v);
if(x!=y)
{
ans[Qu[i].num]+=heav[x]*heav[y];
// cout<<Qu[i].num<<" # "<<ans[Qu[i].num]<<endl;
Union(x,y);
}
j++;
}
}
for(int i=1;i<=q;i++)
{
printf("%d\n",ans[i]);
}
}
int main()
{
while(scanf("%d%d%d",&n,&m,&q)!=EOF)
{
init();
solve();
}
return 0;
}
HDU 3938 离线并查集
最新推荐文章于 2020-06-16 14:42:08 发布