[HDU 5441]Travel：并查集

最新推荐文章于 2019-03-18 20:48:45 发布

原创最新推荐文章于 2019-03-18 20:48:45 发布 · 326 阅读

0 ·

CC 4.0 BY-SA版权

HDU 同时被 2 个专栏收录

16 篇文章

订阅专栏

并查集

5 篇文章

订阅专栏

本文介绍了一种基于离线处理的图算法实现方法，通过预处理所有边和询问，并按x值排序，实现高效的合并联通块操作。文章提供了一个完整的C++代码示例，展示了如何在一系列边和询问中计算联通块合并前后的答案变化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

点击这里查看原题

离线处理，将所有边和询问读入后按x排序，从小到大加边，若不在同一个联通块内则ans加上两个联通块合并后的答案，减去原来的两个块的答案。

/*
User:Small
Language:C++
Problem No.:5441
*/
#include<bits/stdc++.h>
#define ll long long
#define inf 999999999
using namespace std;
int ans[5005],pre[20005],num[20005],n,m,k;
struct no{
    int x,id;
}p[5005];
struct edge{
    int u,v,w;
}e[100005];
void init(){
    for(int i=1;i<=n;i++){
        pre[i]=i;
        num[i]=1;
    }
}
bool cmpw(edge a,edge b){
    return a.w<b.w;
}
bool cmpx(no a,no b){
    return a.x<b.x;
}
int get(int x){
    return x==pre[x]?x:pre[x]=get(pre[x]);
}
void cal(){
    int res=0;
    for(int i=1,j=1;i<=k;i++){
        while(j<=m&&e[j].w<=p[i].x){
            int u=get(e[j].u),v=get(e[j].v);
            if(u!=v){
                res+=(num[u]+num[v])*(num[u]+num[v]-1)-num[u]*(num[u]-1)-num[v]*(num[v]-1);
                pre[v]=u;
                num[u]+=num[v];
            }
            j++;
        }
        ans[p[i].id]=res;
    }
}
void solve(){
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++){
        pre[i]=i;
        num[i]=1;
    }
    for(int i=1;i<=m;i++)
        cin>>e[i].u>>e[i].v>>e[i].w;
    for(int i=1;i<=k;i++){
        cin>>p[i].x;
        p[i].id=i;
    }
    sort(e+1,e+m+1,cmpw);
    sort(p+1,p+k+1,cmpx);
    cal();
    for(int i=1;i<=k;i++)
        cout<<ans[i]<<endl;
}
int main(){
    freopen("data.in","r",stdin);//
    ios::sync_with_stdio(false);
    int t;
    cin>>t;
    while(t--) solve();
    return 0;
}