HDU5441 Travel

1.题目描述：给你n点m边的网络和Q和流量限制，求每走一步都满足流量限制下，可连通的点对。

2.解题思路：开始看有点网络流的意思，但是并没有源点汇点。因为是双向边，从连通方面想，假设有两个点集a、b，点集中边都满足<=limit，那么只要连接这两个点集的任意一条边<=limit，那么这两个点集就可以合成新的点集，并且增加的点对个数是a点数*b点数。那么回到最初始的状态就是2个点1条边。也就是说这是个并查集的问题，一直往里加边即可。
那么问题来了，询问5000次，一次并查集20W，乘起来肯定超时，但可以发现并查集只要做一次就够了，把limit和边都按从小到大排序，只要边权大于当前limit就记录当前limit的结果，直到边全部处理完即可

3.代码：

#include<iostream>
#include<fstream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#define ms(x) memset(x,0,sizeof(x))
#define FOR(i,s,t) for(int i=(s);i<(t);i++)
#define CFOR(i) for(;i>0;i--)
#define DFOR(i,s,t) for(int i=(s);i>=(t);i--)
#define PFOR(i,s,t,p) for(int i=(s);i<=(t);i+=p)
#define SD3(i,j,l) scanf("%d%d%d",&i,&j,&l);
#define SLLD(i) scanf("%lld",&i)
#define SD(i) scanf("%d",&i)
#define MEM(a,t) memset(a,t,sizeof(a))
#define PIA system("pause")
#define MAXINT 0xfffffff
#define maxn 100000+10
#define MARK cout<<"MARK"<<endl;
#define BUG cout<<"error!"<<endl;
#define ll long long
using namespace std;
int fa[maxn];
ll num[maxn];
ll ans[8000];
struct edge{
   int a,b,v;
}e[maxn];
struct que{
    int x,id;
}qe[8000];
int t,n,m,q;
int fin(int x)
{
    if(fa[x]!=x)
        fa[x]=fin(fa[x]);
    return fa[x];
}
bool cmp_e(edge k1,edge k2)
{
    return k1.v<k2.v;
}
bool cmp_q(que k1,que k2)
{
    return k1.x<k2.x;
}

void init()
{
    int i;
    for(i=0;i<n;i++)
        {fa[i]=i;num[i]=1;}
     for(i=0;i<m;i++)
            SD3(e[i].a,e[i].b,e[i].v);
        for(i=0;i<q;i++)
        {
            SD(qe[i].x),qe[i].id=i;
        }
        sort(e,e+m,cmp_e);
        sort(qe,qe+q,cmp_q);
}
int main()
{

    cin>>t;
    while(t--)
    {
        SD3(n,m,q);
        init();
        int i=0,j=0;
        ll temp=0;
        while(j<q)
        {
            for(;i<m&&e[i].v<=qe[j].x;i++)
            {
                int p1=fin(e[i].a),p2=fin(e[i].b);
                if(p1!=p2)
                {
                    temp+=num[p1]*num[p2];
                 //   printf("temp:%d\n",num[p1]*num[p2]);
                    if(p1>p2)
                        swap(p1,p2);
                    num[p1]+=num[p2];
                    fa[p2]=p1;
                }
            }
            ans[qe[j].id]=temp*2;
            j++;
        }
        for(i=0;i<q;i++)
            printf("%lld\n",ans[i]);
    }
    return 0;
}