HDU 5723 Abandoned country(最小生成树 + 树形DP)

forever97 posted @ 2016年7月21日 23:12 in 图论-最小生成树 with tags 最小生成树 树形DP 期望 , 499 阅读

 

【题目链接】 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\n",ans1,ans2/n/(n-1));
    }return 0;
}

 


登录 *


loading captcha image...
(输入验证码)
or Ctrl+Enter