[并查集][排序] Jzoj P4223 旅游

Description

 

Input

Output

 

Sample Input

1
5 5 3
2 3 6334
1 5 15724
3 5 5705
4 3 12382
1 3 21726
6000
10000
13000

Sample Output

2
6
12

 

Data Constraint

 

题解

• 题目大意：有n个点，m条边，每条边有一个权值，每次问有多少个点对在不经过权值超过x的边可以到达
• 这题的话，可以将询问离线来做，将边权和询问权值从小到大排序
• 然后将边从小到大打进并查集里，当前边的贡献就是边所连的两点之间的子树大小的乘积*2的和

代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#define N 200010
#define ll long long
using namespace std;
ll T,n,m,q,sum,l,fa[N],size[N],ans[N];
struct edge { ll x,y,d; }e[N],Q[N];
bool cmp(edge x,edge y) { return x.d<y.d; }
bool cmp1(edge x,edge y) { return x.x<y.x; }
ll getfather(ll x) {return !fa[x]?x:fa[x]=getfather(fa[x]);}
int main()
{
    scanf("%lld",&T);
    while (T--)
    {
        scanf("%lld%lld%lld",&n,&m,&q),l=1,sum=0,memset(ans,0,sizeof(ans)),memset(fa,0,sizeof(fa));
        for (ll i=1;i<=n;i++) size[i]=1;
        for (ll i=1;i<=m;i++) scanf("%lld%lld%lld",&e[i].x,&e[i].y,&e[i].d);
        for (ll i=1;i<=q;i++) scanf("%lld",&Q[i].x),Q[i].d=i;
        sort(e+1,e+m+1,cmp),sort(Q+1,Q+q+1,cmp1);
        for (ll i=1;i<=q;i++)
        {
            while (l<m&&e[l].d<=Q[i].x)
            {
                ll u=getfather(e[l].x),v=getfather(e[l].y);
                if (u!=v) fa[v]=u,sum+=size[u]*size[v]*2,size[u]+=size[v],size[v]=0;
                l++;
            }
            ans[Q[i].d]=sum;
        }
        for (ll i=1;i<=q;i++) printf("%lld\n",ans[i]);
    }
}