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+;

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

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

 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=; i<m; i++ )
         scanf("%d%d%d", &es[i].from, &es[i].to, &es[i].cost );
     sort( es, es+m );
     for( int i=; i<q; i++ )    {
         scanf("%d", &que[i].val );
         que[i].id = i;
     }
     sort( que, que+q );
     for( int i=; i<=n; i++ )   { pre[i] = i; sum[i] = ; }
     int j = ;
     int res = ;
     for( int i=; 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=; i<q; i++ )
         printf("%d\n", ans[i]<< );
 }

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

     return ;
 }