hdu 5441

这道题目是用并查集来维护联通点的个数。
我们首先将所有路径按权值大小从小到大排序，然后将所有询问也按权值从小到大排序（没错，我们要离线处理），接着遍历一遍询问，将符合条件的路径两端点合并，我们用rk[i]来代表i点所在并查集的大小，假设两个端点分别为u,v，那么它们所在并查集大小为rk[u],rk[v],则当这两个点合并以后，这两个并查集中的点就合并起来了，根据题意，多出来的数目为rk[u]*rk[v]*2。
对了，刚开始的路径压缩写挫了。。T了好多发。。

#include<stdio.h>
#include<iostream>
#include<string>
#include<string.h>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<time.h>
#include<queue>
#include<stack>
#include<iterator>
#include<math.h>
#include<stdlib.h>
#include<limits.h>
#include<map>
#include<set>
#include<bitset>
//#define ONLINE_JUDGE
#define eps 1e-5
#define INF 0x7fffffff
#define FOR(i,a) for((i)=0;i<(a);(i)++)
#define MEM(a) (memset((a),0,sizeof(a)))
#define sfs(a) scanf("%s",a)
#define sf(a) scanf("%d",&a)
#define sfI(a) scanf("%I64d",&a)
#define pf(a) printf("%d\n",a)
#define pfI(a) printf("%I64d\n",a)
#define pfs(a) printf("%s\n",a)
#define sfd(a,b) scanf("%d%d",&a,&b)
#define sft(a,b,num) scanf("%d%d%d",&a,&b,&num)
#define for1(i,a,b) for(int i=(a);i<b;i++)
#define for2(i,a,b) for(int i=(a);i<=b;i++)
#define for3(i,a,b)for(int i=(b);i>=a;i--)
#define MEM1(a) memset(a,0,sizeof(a))
#define MEM2(a) memset(a,-1,sizeof(a))
#define ll __int64
const double PI=acos(-1.0);
template<class T> T gcd(T a,T b){return b?gcd(b,a%b):a;}
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
template<class T> inline T Min(T a,T b){return a<b?a:b;}
template<class T> inline T Max(T a,T b){return a>b?a:b;}
using namespace std;
//#pragma comment(linker,"/STACK:1024000000,1024000000")
int n,m,q;
#define M 1110
#define N 100010
#define Mod 1000000000
#define p(x,y) make_pair(x,y)
const int MAX_len=550;
vector<int>G[N];
int ans[N];
int res;
int fa[N];
int rk[N];
struct Node{
    int u,v,val;
    bool operator <(const Node &x)const{    //将路径按权值大小从小到大排序
        return val<x.val;
    }
}e[N];
struct Que{
    int val,id;
    bool operator <(const Que &x)const{ //将询问按权值大小排序
        return val<x.val;
    }
}que[N];
void addEdge(int p,int u,int v,int val){
    e[p].u = u;
    e[p].v = v;
    e[p].val = val;
}
int findfa(int x){
    return x==fa[x]?x:fa[x]=findfa(fa[x]); //要写成fa[x]=findfa(fa[x])，这样子更快，否则会T
}
void Union(int x,int y){
    int fx = findfa(x);
    int fy = findfa(y);
    if(fx != fy){
        fa[fx] = fy;
        int num1 = rk[fx];
        int num2 = rk[fy];
        rk[fy] += rk[fx];       //按秩压缩
        res += num1*num2*2;     //合并
    }
}
void Deal(){
    int qr=0;
    for(int i=0;i<q;i++){
        while(qr<m && e[qr].val<=que[i].val){   //符合条件的点
            int u = e[qr].u;
            int v = e[qr].v;
            Union(u,v);
            qr++;
        }
        ans[que[i].id] = res;
    }
}
int main(){
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
//  freopen("out.txt", "w", stdout);
#endif
    int T;
    sf(T);
    while(T--){
        scanf("%d%d%d",&n,&m,&q);
        for(int i=0;i<=m;i++)G[i].clear();
        int u,v,d;
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&u,&v,&d);
            addEdge(i,u,v,d);
        }
        sort(e,e+m);
        for(int i=0;i<q;i++){
            sf(que[i].val);
            que[i].id = i;
        }
        sort(que,que+q);
        res=0;
        for(int i=0;i<=n;i++){
            fa[i]=i;
            rk[i]=1;
        }
        memset(ans,0,sizeof ans);
        Deal();
        for(int i=0;i<q;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}