【题目链接】 http://acm.hdu.edu.cn/showproblem.php?pid=5723
【题目大意】
n座城市,m条路径,求解:
1.最短的路径和,使得n座城市之间直接或者间接连通
2.在路径和最短的情况下,求出任意两个城市之间的期望距离
【题解】
对于问题1,只需求出该图的最小生成树,边权和即答案,由于边权值唯一,因此不存在最小生成树多解的情况。
对于问题2,期望的通常求法为(任意两点之间的路径和)/(点对数)
那么问题就转化为任意两点间距离和的问题,我们按边考虑,对于每条树上的边,它对答案的贡献值为左边的点数×右边的点数×边权,搜索每个记录每棵子树的大小,对于子树和父节点相连的这条边,他左右两边的点数分别为(总点数-子树大小)和(子树大小),那么递归计算答案即可。
【代码】
#include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int N=1000005; int T,n,m,f[N],nxt[N],w[N],v[N],cnt[N],g[N],ed; double ans1,ans2; struct data{int x,y,z;}a[N]; bool cmp(data a,data b){return a.z<b.z;} int sf(int x){return x==f[x]?x:f[x]=sf(f[x]);} void add(int x,int y,int z){v[++ed]=y;w[ed]=z;nxt[ed]=g[x];g[x]=ed;} void dfs(int x,int pre){ cnt[x]=1; for(int i=g[x];i;i=nxt[i])if(v[i]!=pre){ dfs(v[i],x); cnt[x]+=cnt[v[i]]; ans2+=2.0*cnt[v[i]]*(n-cnt[v[i]])*w[i]; } } int main(){ scanf("%d",&T); while(T--){ ans1=ans2=ed=0; memset(v,0,sizeof(v)); memset(nxt,0,sizeof(nxt)); memset(w,0,sizeof(w)); memset(g,0,sizeof(g)); scanf("%d%d",&n,&m); for(int i=1;i<=m;i++)scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z); sort(a+1,a+m+1,cmp); for(int i=1;i<=n;i++)f[i]=i; for(int i=1;i<=m;i++){ if(sf(a[i].x)==sf(a[i].y))continue; f[sf(a[i].x)]=sf(a[i].y); ans1+=a[i].z; add(a[i].x,a[i].y,a[i].z); add(a[i].y,a[i].x,a[i].z); }dfs(1,1); printf("%.0f %.2f ",ans1,ans2/n/(n-1)); }return 0; }