題意：

給出一個n個點、m條邊的無向圖，邊上有權值，有q組詢問，每組詢問給出一個數字x，我們要在圖中找出‘點對’的個數，這些‘點對’（例如a，b）滿足從a到b有一條路徑經過的每一條邊都要小于x，輸出每組詢問點對的數量。

思路：

并查集。

我們把查詢操作離線化，把邊按權值從小到大排序，把查詢操作也從小到大排序，每次查詢操作都掃一遍邊，判斷滿足的邊的端點在不在同一個并查集中，如果不在就合并這兩個并查集，同時記錄新增加的數量，掃到不滿足的邊就跳出，繼續下一次查詢操作。

代碼：

#include<cstdio> #include<cstring> #include<iostream> #include<algorithm>using namespace std;struct Edge {int u,v,x; }e[100005];int cmp(Edge a,Edge b) {return a.x<b.x; }struct Q {int pos,x; }query[5005];int cmp1(Q a,Q b) {return a.x<b.x; }int ans[5005]; int fa[200005]; int num[200005]; int n,m,q;void init() {for(int i=0;i<=n;i++){fa[i]=i;num[i]=1;}return ; }int find1(int x) {return fa[x]==x? x: fa[x]=find1(fa[x]); }void union1(int x,int y) {x=find1(x);y=find1(y);if(y<x)swap(x,y);fa[y]=x;num[x]+=num[y];return ; }int main() {int t;scanf("%d",&t);while(t--){scanf("%d%d%d",&n,&m,&q);for(int i=0;i<m;i++)scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].x);init();for(int i=0;i<q;i++){scanf("%d",&query[i].x);query[i].pos=i;}sort(e,e+m,cmp);sort(query,query+q,cmp1);int now=0;int tmp=0;for(int i=0;i<q;i++){// printf("now=%d\n",now);int j;for(j=now;j<m;j++){if(e[j].x>query[i].x)break;int fax=find1(e[j].u);int fay=find1(e[j].v);//printf("fax=%d fay=%d\n",fax,fay);if(fax==fay)continue;else{int tmp1=(num[fax]+num[fay])*(num[fax]+num[fay]-1);int tmp2=(num[fax]-1)*num[fax];int tmp3=(num[fay]-1)*num[fay];tmp+=(tmp1-tmp2-tmp3);union1(fax,fay);}}now=j;ans[query[i].pos]=tmp;}for(int i=0;i<q;i++)printf("%d\n",ans[i]);}return 0; }

總結

以上是生活随笔為你收集整理的hdu5441(2015长春网络赛E题)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。