http://acm.hdu.edu.cn/showproblem.php?pid=5441
题意:
给出一个图,每条边有一个距离,现在有多个询问,每个询问有一个距离值d,对于每一个询问,计算出有多少点对(x,y)使得在x到y的路径上没有一条边的距离大于d。
思路:
只要边距离小于d,那就是可行的,直接加入并查集来维护。并查集需要维护一下树的节点个数。
将边和询问都排序离线处理。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll; int n,m,k;
int p[], num[];
ll ans[]; struct node
{
int u,v,w;
bool operator< (const node& rhs) const
{
return w<rhs.w;
}
}e[]; struct Node
{
int id;
int val;
bool operator< (const Node& rhs) const
{
return val<rhs.val;
}
}query[]; int finds(int x)
{
return x==p[x]?x:p[x] = finds(p[x]);
} int main()
{
//freopen("in.txt","r",stdin);
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d%d",&n,&m,&k);
for(int i=;i<=n;i++) {p[i] = i; num[i] = ;}
for(int i=;i<=m;i++)
scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
sort(e+,e+m+);
for(int i=;i<=k;i++)
{
scanf("%d",&query[i].val);
query[i].id = i;
}
sort(query+,query+k+);
query[].id = ;
ll tmp = ;
int pos = ;
ans[] = ;
for(int i=;i<=m && pos<=k;i++)
{
while(e[i].w>query[pos].val && pos<=k)
{
ans[query[pos].id] = ans[query[pos-].id];
pos++;
}
while(e[i].w<=query[pos].val && i<=m)
{
int x = finds(e[i].u);
int y = finds(e[i].v);
if(x!=y)
{
p[x] = y;
tmp = tmp-num[y]*(num[y]-)-num[x]*(num[x]-);
num[y] += num[x];
tmp = tmp+num[y]*(num[y]-);
}
i++;
}
ans[query[pos].id] = tmp;
pos++;
i--;
}
while(pos<=k)
{
ans[query[pos].id] = ans[query[pos-].id];
pos++;
}
for(int i=;i<=k;i++) printf("%lld\n",ans[i]);
}
return ;
}