题意描述:简单的讲就是,给你一张无向图,求有多少条路径使得路径上的花费小于L,这里路径上的花费是这样规定的,a、b两点之间的多条路径中的最长的边最小值!
题解:用在这题中,是因为输入中的“询问部分”,有Q 个问,每个L可以有多少种不同路径。由于大的L必定会包含到小的L, 所以把所有问题都输入,再从大到小排序,再计算,可以减少很多计算量。
这题还需要用到的是并查集中的“权值”, 用rank数组表示,也就是某个棵树k有rank【k】个结点。同一个树之间的点都是连通的,任何点都可以通往其它的任意点, 那么当两颗树合并成一棵树时, 将会增加rank[a]*rank[b]条路径。
#include <set>
#include <map>
#include <queue>
#include <deque>
#include <cmath>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cassert>
#include <iostream>
#include <algorithm>
using namespace std;
#define L(i) i<<1
#define R(i) i<<1|1
#define INF 0x3f3f3f3f
#define pi acos(-1.0)
#define eps 1e-9
#define maxn 1000010
#define MOD 1000000007
#define zero(x) (((x)>0?(x):-(x))<eps)
int ans,fa[10010],s[10010];
int n,m,q;
int sum_weight;
map<pair<int,int>,int> mp;
struct node
{
int u,v,val;
bool operator < (const node & a)const
{
return val < a.val;
}
}cnt[50010];
struct query
{
int length,id;
bool operator < (const query & a)const
{
return length < a.length;
}
}Query[10010];
int Find(int x)
{
return x == fa[x]?x:fa[x] = Find(fa[x]);
}
int Union(int x,int y)
{
int px = Find(x);
int py = Find(y);
if(px == py)
return 0;
else
{
int k = s[px] * s[py];
s[py] += s[px];
s[px] = 0;
fa[px] = py;
return k;
}
}
//void Kruskal()
//{
// sum_weight = 0;
// for(int i = 0; i < m; i++)
// {
// int u = edge[i].u;
// int v = edge[i].v;
// if(Find(u) != Find(v))
// {
// Union(u,v);
// sum_weight += edge[i].w;
// printf("%d %d %d\n",u,v,sum_weight);
// }
// }
// printf("%d\n",sum_weight);
//}
void init()
{
for(int i = 0; i <= n; i++)
{
fa[i] = i;
s[i] = 1;
}
}
int main()
{
int t,C = 1;
//scanf("%d",&t);
while(scanf("%d%d%d",&n,&m,&q) != EOF)
{
init();
for(int i = 0; i < m; i++)
scanf("%d%d%d",&cnt[i].u,&cnt[i].v,&cnt[i].val);
for(int i = 0; i < q; i++)
{
Query[i].id = i;
scanf("%d",&Query[i].length);
}
sort(cnt,cnt+m);
sort(Query,Query+q);
int k = 0,ans[10010] = {0};
int num = 0;
for(int i = 0; i < q; i++)
{
while(k < m && cnt[k].val <= Query[i].length)
{
num += Union(cnt[k].u,cnt[k].v);
k++;
}
ans[Query[i].id] = num;
}
for(int i = 0; i < q; i++)
printf("%d\n",ans[i]);
}
return 0;
}