离线+生成树+并查集——cf1213G

#include<bits/stdc++.h>
using namespace std;
#define N 200005
#define ll long long
struct Edge{int u,v,w;}e[N];
struct Query{int id,qq;}qq[N];
int n,q;
ll ans[N],sum;
bool operator<(Edge a,Edge b){return a.w<b.w;}
bool operator<(Query a,Query b){return a.qq<b.qq;}

int F[N],size[N];
int find(int x){
    if(F[x]==x)return x;
    int f=find(F[x]);
    return F[x]=f;
}
void bing(int u,int v){
    int f1=find(u),f2=find(v);
    if(f1==f2)return;
    sum-=(ll)size[f1]*(size[f1]-)/;
    sum-=(ll)size[f2]*(size[f2]-)/;
    size[f1]+=size[f2];
    sum+=(ll)size[f1]*(size[f1]-)/;
    F[f2]=f1;
}

int main(){
    cin>>n>>q;
    for(int i=;i<n;i++)
        scanf("%d%d%lld",&e[i].u,&e[i].v,&e[i].w);
    for(int i=;i<=q;i++)
        scanf("%d",&qq[i].qq),qq[i].id=i;
    sort(e+,e+n);
    sort(qq+,qq++q);

    for(int i=;i<=n;i++)F[i]=i,size[i]=;

    int p=;
    for(int i=;i<=q;i++){
        while(qq[i].qq>=e[p].w && p<=n-){
            int u=e[p].u,v=e[p].v;
            bing(u,v);
            p++;
        }
        ans[qq[i].id]=sum;
    }
    for(int i=;i<=q;i++)
        cout<<ans[i]<<" ";
}