HDU OJ 5441 Travel 2015online E

题目：click here

题意：

　　有个很暴躁的人，想坐车旅行n个城市。连接城市共有m条路（双向）。他坐在车上很不爽，每次最多忍耐x分钟。但是每站下车他又可以休息（重新计时）。总共有q次询问。问途中有多少条路他可以不爆发。a到b 和 b到a 算不同的路。 a 和 b 必须不相同。 

分析：

　　并查集

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;
const int M = 2e4+5;

struct Edge {   // 保存结点信息
    int from, to, cost;
    bool operator < ( const Edge x ) const { return cost < x.cost; }
} es[M*5];

struct Query    {   //  保存问题信息
    int val, id;
    bool operator < ( const Query x ) const { return val < x.val; };
} que[5005];

int n, m, q;
int pre[M];     // 并查集
int ans[M];     // 答案
int sum[M];     // 记录并查集树高度

int find( int x )   {
    return pre[x] == x ? x : pre[x] = find( pre[x] );
}
void merge( int x, int y )  {
    x = find( x );  y = find( y );
    if( sum[x] > sum[y] )   { sum[x] += sum[y]; pre[y] = x; }
    else    { sum[y] += sum[x]; pre[x] = y; }
}
void solve()    {
    scanf("%d%d%d", &n, &m, &q );
    for( int i=0; i<m; i++ )
        scanf("%d%d%d", &es[i].from, &es[i].to, &es[i].cost );
    sort( es, es+m );
    for( int i=0; i<q; i++ )    {
        scanf("%d", &que[i].val );
        que[i].id = i;
    }
    sort( que, que+q );
    for( int i=0; i<=n; i++ )   { pre[i] = i; sum[i] = 1; }
    int j = 0;
    int res = 0;
    for( int i=0; i<q; i++ )    {
        for( ; j<m && que[i].val>=es[j].cost; j++ ) {
            int x = find( es[j].from );
            int y = find( es[j].to );
            if( x == y )    continue;
            res += sum[x] * sum[y];
            merge( es[j].from, es[j].to );
        }
        ans[que[i].id] = res;
    }
    for( int i=0; i<q; i++ )
        printf("%d\n", ans[i]<<1 );
}

int main()  {
    int t;  scanf("%d", &t );
    while( t-- )    {
        solve();        
    }

    return 0;
}