hdu 3938(离线的并查集）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3938

思路：对于边进行从小到大进行排序，题目要求a、b两点之间所有的路径中的最大边的最小值，在满足边的大小<=len的条件下，令r1=Find(a),r2=Find(b),若r1==r2,此时就是简单的合并，并且更新一下集合的个数（abs(parent[i])即为以i为根结点的集合的大小），若r1!=r2，此时a,b两点之间的路径的条数即为parent[r1]*parent[r2]。显然对于那些边>=len的，都要加上原先的路径数目。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 10000+100
#define MAXM 50000+555
typedef long long LL;
int parent[MAXN];
int n,m,q;
LL path[MAXN];
struct Edge{
int u,v,w;
}edge[MAXM];

struct Node{
int id,len;
}p[MAXN];

int Find(int x)
{
int s;
for(s=x;parent[s]>=0;s=parent[s])
;
while(x!=s){
int tmp=parent[x];
parent[x]=s;
x=tmp;
}
return s;
}

int  Union(int u,int v)
{
int r1=Find(u),r2=Find(v);
if(r1==r2)return 0;
int tmp=parent[r1]*parent[r2];
if(parent[r1]<parent[r2]){
parent[r1]+=parent[r2];
parent[r2]=r1;
}else {
parent[r2]+=parent[r1];
parent[r1]=r2;
}
return tmp;
}

int cmp1(const Edge &a,const Edge &b)
{
return a.w<b.w;
}

int cmp2(const Node &a,const Node &b)
{
return a.len<b.len;
}

int main()
{
while(~scanf("%d%d%d",&n,&m,&q)){
memset(parent,-1,sizeof(int)*(n+2));
for(int i=0;i<m;i++){
scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
}
sort(edge,edge+m,cmp1);
for(int i=0;i<q;i++){
scanf("%d",&p[i].len);
p[i].id=i;
}
sort(p,p+q,cmp2);
LL ans=0;
int pos=0;
for(int i=0;i<q;i++)
{
while(pos<m&&edge[pos].w<=p[i].len){
ans+=Union(edge[pos].u,edge[pos].v);
pos++;
}
path[p[i].id]=ans;
}
for(int i=0;i<q;i++)
printf("%I64d\n",path[i]);
}
return 0;
}

View Code