【JZOJ 3886】 道路维护

Description

最近徆多人投诉说C国的道路破损程度太大，以至亍无法通行
C国的政府徆重视这件事，但是最近财政有点紧，丌可能将所有的道路都进行维护，所以他们决定按照下述方案进行维护
将C国抽象成一个无向图，定义两个城市乊间的某条路径的破损程度为该条路径上所有边破损程度的最大值，定义两个城市乊间的破损程度为两个城市乊间所有路径破损程度的最小值
然后C国政府向你提问多次，有多少个城市对的破损程度丌超过L，他们将依照你的回答来决定到底怎样维护C国的道路

100%数据满足n≤10^4，m，q≤10^5，0≤c，Li≤10^9

Analysis

这题很水，我比赛时切了
离线，询问L动态加入比L小的边，维护并查集计算加入一条边增加的连通点对数

Code

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N=10010,M=100010;
struct edge
{
    int x,y,z;
}a[M];
struct ask
{
    int x,id;
}b[M];
int n,m,q,ans[M],f[N],sum[N];
bool cmp(edge a,edge b)
{
    return a.z<b.z;
}
bool cmpask(ask a,ask b)
{
    return a.x<b.x;
}
int getfa(int v)
{
    if(f[v]==v) return v;
    sum[f[v]]+=sum[v];
    sum[v]=0;
    return f[v]=getfa(f[v]);
}
int main()
{
    scanf("%d %d %d",&n,&m,&q);
    fo(i,1,m) scanf("%d %d %d",&a[i].x,&a[i].y,&a[i].z);
    sort(a+1,a+m+1,cmp);
    fo(i,1,q) scanf("%d",&b[i].x),b[i].id=i;
    sort(b+1,b+q+1,cmpask);
    fo(i,1,n) f[i]=i,sum[i]=1;
    int j=1,tot=0;
    fo(i,1,q)
    {
        int x=b[i].x;
        while(a[j].z<=x && j<=m)
        {
            int u=a[j].x,v=a[j].y;
            int fu=getfa(u),fv=getfa(v);
            if(fu!=fv)
            {
                tot+=sum[fu]*sum[fv];
                f[fv]=fu;
                getfa(fv);
            }
            j++;
        }
        ans[b[i].id]=tot;
    }
    fo(i,1,q) printf("%d ",ans[i]);
    return 0;
}